Optimized / scheduled graph ready to run with feeds. 更多...

#include <Graph.h>

Public 成员函数
	CompiledFunction ()

	~CompiledFunction ()

Tensor *	run0 ()

Tensor *	run1 (Tensor *in0)

Tensor *	run2 (Tensor in0, Tensor in1)

Tensor *	run3 (Tensor in0, Tensor in1, Tensor *in2)

Tensor *	run4 (Tensor in0, Tensor in1, Tensor in2, Tensor in3)

Tensor *	run5 (Tensor in0, Tensor in1, Tensor in2, Tensor in3, Tensor *in4)

Tensor *	run6 (Tensor in0, Tensor in1, Tensor in2, Tensor in3, Tensor in4, Tensor in5)

int	getPlaceholderCount () const

std::string	getDevice () const

静态 Public 成员函数
static CompiledFunction *	fromFunc (Func *fn)

详细描述

Optimized / scheduled graph ready to run with feeds.

在文件 Graph.h 第 191 行定义.

构造及析构函数说明

◆ CompiledFunction()

eve::tensor::CompiledFunction::CompiledFunction ( )

default

被这些函数引用 fromFunc().

◆ ~CompiledFunction()

eve::tensor::CompiledFunction::~CompiledFunction ( )

default

成员函数说明

◆ fromFunc()

CompiledFunction * eve::tensor::CompiledFunction::fromFunc ( Func * fn )

static

在文件 Graph.cpp 第 590 行定义.

引用了 CompiledFunction(), fn, eve::tensor::optimizeGraph() , 以及 eve::tensor::GpuProgram::tryBuild().

被这些函数引用 eve::tensor::Func::compile().

◆ getDevice()

std::string eve::tensor::CompiledFunction::getDevice ( ) const

inline

在文件 Graph.h 第 205 行定义.

◆ getPlaceholderCount()

int eve::tensor::CompiledFunction::getPlaceholderCount ( ) const

inline

在文件 Graph.h 第 204 行定义.

◆ run0()

Tensor * eve::tensor::CompiledFunction::run0 ( )

在文件 Graph.cpp 第 611 行定义.

◆ run1()

Tensor * eve::tensor::CompiledFunction::run1 ( Tensor * in0 )

在文件 Graph.cpp 第 613 行定义.

◆ run2()

Tensor * eve::tensor::CompiledFunction::run2	(	Tensor *	in0,
		Tensor *	in1
	)

在文件 Graph.cpp 第 618 行定义.

◆ run3()

Tensor * eve::tensor::CompiledFunction::run3	(	Tensor *	in0,
		Tensor *	in1,
		Tensor *	in2
	)

在文件 Graph.cpp 第 623 行定义.

◆ run4()

Tensor * eve::tensor::CompiledFunction::run4	(	Tensor *	in0,
		Tensor *	in1,
		Tensor *	in2,
		Tensor *	in3
	)

在文件 Graph.cpp 第 628 行定义.

◆ run5()

Tensor * eve::tensor::CompiledFunction::run5	(	Tensor *	in0,
		Tensor *	in1,
		Tensor *	in2,
		Tensor *	in3,
		Tensor *	in4
	)

在文件 Graph.cpp 第 633 行定义.

◆ run6()

Tensor * eve::tensor::CompiledFunction::run6	(	Tensor *	in0,
		Tensor *	in1,
		Tensor *	in2,
		Tensor *	in3,
		Tensor *	in4,
		Tensor *	in5
	)

在文件 Graph.cpp 第 638 行定义.

该类的文档由以下文件生成:

src/modules/tensor/Graph.h
src/modules/tensor/Graph.cpp

Public 成员函数

静态 Public 成员函数