eve::tensor::Func类 参考

Trace builder — TF2 tf.function analogue (tf.func in scripts). While active, TF ops record into this graph. 更多...

#include <Graph.h>

Public 成员函数
	Func (TF *owner)
	~Func ()
Tensor *	input1 (int d0)
Tensor *	input2 (int d0, int d1)
Tensor *	input3 (int d0, int d1, int d2)
Tensor *	input4 (int d0, int d1, int d2, int d3)
Tensor *	input5 (int d0, int d1, int d2, int d3, int d4)
Tensor *	input6 (int d0, int d1, int d2, int d3, int d4, int d5)
void	setOutput (Tensor *t)
class CompiledFunction *	compile ()
Graph &	graph ()
const Graph &	graph () const
TF *	owner () const
bool	isTracing () const
int	outputNode () const
int	placeholderCount () const
int	ensureNode (const Tensor *t)
	Ensure tensor is a node in this graph (Const-capture if eager).
Tensor *	emitUnary (OpType type, const Tensor *x)
Tensor *	emitUnaryScalar (OpType type, const Tensor *x, float s0, float s1=0.f)
Tensor *	emitBinary (OpType type, const Tensor *a, const Tensor *b)
Tensor *	emitTernary (OpType type, const Tensor *a, const Tensor *b, const Tensor *c)
Tensor *	emitMatMul (const Tensor *a, const Tensor *b)
Tensor *	emitTranspose (const Tensor *x)
Tensor *	emitPermute (const Tensor *x, const int *order, int rank)
Tensor *	emitReshape (const Tensor *x, const int *dims, int rank)
Tensor *	emitFill (const int *dims, int rank, float value)
Tensor *	emitSoftmax (const Tensor *x, int axis, bool logMode)
Tensor *	emitLayerNorm (const Tensor *x, const Tensor *scale, const Tensor *bias, float eps)
Tensor *	emitRMSNorm (const Tensor *x, const Tensor *scale, float eps)
Tensor *	emitConv1d (const Tensor *x, const Tensor *w, const Tensor *bias, int stride, int pad)
Tensor *	emitConv2d (const Tensor *x, const Tensor *w, const Tensor *bias, int stride, int pad)
Tensor *	emitPool (OpType type, const Tensor *x, int ksize, int stride, int pad)
Tensor *	emitEmbedding (const Tensor *table, const Tensor *indices)
Tensor *	emitConcat (const Tensor *const *ins, int n, int axis)
Tensor *	emitSlice (const Tensor *x, int axis, int begin, int end)
Tensor *	emitReduce (OpType type, const Tensor *x, int axis, bool keepDims)
Tensor *	emitArgMax (const Tensor *x, int axis, bool keepDims)
Tensor *	emitCast (const Tensor *x, DType dtype)
Tensor *	emitSdpa (const Tensor *q, const Tensor *k, const Tensor *v, const Tensor *mask, float scale)
Tensor *	emitResize2d (const Tensor *x, int outH, int outW, int mode)

详细描述

Trace builder — TF2 tf.function analogue (tf.func in scripts). While active, TF ops record into this graph.

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

构造及析构函数说明

Func()

eve::tensor::Func::Func ( TF *  owner )

explicit

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

引用了 eve::tensor::TF::pushTrace().

~Func()

eve::tensor::Func::~Func

(

)

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

引用了 eve::tensor::TF::popTrace().

成员函数说明

compile()

CompiledFunction * eve::tensor::Func::compile

(

)

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

引用了 eve::tensor::CompiledFunction::fromFunc() , 以及 eve::tensor::TF::popTrace().

emitArgMax()

Tensor * eve::tensor::Func::emitArgMax	(	const Tensor *	x,
		int	axis,
		bool	keepDims
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::ArgMax, eve::tensor::GraphNode::dims, ensureNode(), eve::tensor::Int32, eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank , 以及 x.

emitBinary()

Tensor * eve::tensor::Func::emitBinary	(	OpType	type,
		const Tensor *	a,
		const Tensor *	b
	)

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

引用了 a, eve::tensor::Graph::addNode(), b, eve::tensor::kernels::broadcastShape(), ensureNode(), eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node() , 以及 type.

emitCast()

Tensor * eve::tensor::Func::emitCast	(	const Tensor *	x,
		DType	dtype
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Cast, eve::tensor::GraphNode::dims, ensureNode(), n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank , 以及 x.

emitConcat()

Tensor * eve::tensor::Func::emitConcat	(	const Tensor *const *	ins,
		int	n,
		int	axis
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Concat, eve::tensor::GraphNode::dims, ensureNode(), eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node() , 以及 eve::tensor::GraphNode::rank.

emitConv1d()

Tensor * eve::tensor::Func::emitConv1d	(	const Tensor *	x,
		const Tensor *	w,
		const Tensor *	bias,
		int	stride,
		int	pad
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Conv1d, ensureNode(), n, eve::tensor::Graph::node(), w , 以及 x.

emitConv2d()

Tensor * eve::tensor::Func::emitConv2d	(	const Tensor *	x,
		const Tensor *	w,
		const Tensor *	bias,
		int	stride,
		int	pad
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Conv2d, ensureNode(), n, eve::tensor::Graph::node(), w , 以及 x.

emitEmbedding()

Tensor * eve::tensor::Func::emitEmbedding	(	const Tensor *	table,
		const Tensor *	indices
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Embedding, ensureNode(), eve::tensor::Tensor::kMaxRank, n , 以及 eve::tensor::Graph::node().

emitFill()

Tensor * eve::tensor::Func::emitFill	(	const int *	dims,
		int	rank,
		float	value
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Const, n , 以及 value.

emitLayerNorm()

Tensor * eve::tensor::Func::emitLayerNorm	(	const Tensor *	x,
		const Tensor *	scale,
		const Tensor *	bias,
		float	eps
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), eve::tensor::LayerNorm, n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, scale , 以及 x.

emitMatMul()

Tensor * eve::tensor::Func::emitMatMul	(	const Tensor *	a,
		const Tensor *	b
	)

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

引用了 a, eve::tensor::Graph::addNode(), b, ensureNode(), eve::tensor::MatMul, n , 以及 eve::tensor::Graph::node().

emitPermute()

Tensor * eve::tensor::Func::emitPermute	(	const Tensor *	x,
		const int *	order,
		int	rank
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node(), eve::tensor::Permute , 以及 x.

被这些函数引用 emitTranspose().

emitPool()

Tensor * eve::tensor::Func::emitPool	(	OpType	type,
		const Tensor *	x,
		int	ksize,
		int	stride,
		int	pad
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, type , 以及 x.

emitReduce()

Tensor * eve::tensor::Func::emitReduce	(	OpType	type,
		const Tensor *	x,
		int	axis,
		bool	keepDims
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, type , 以及 x.

emitReshape()

Tensor * eve::tensor::Func::emitReshape	(	const Tensor *	x,
		const int *	dims,
		int	rank
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), n, eve::tensor::Graph::node(), eve::tensor::Graph::product(), eve::tensor::Reshape , 以及 x.

emitResize2d()

Tensor * eve::tensor::Func::emitResize2d	(	const Tensor *	x,
		int	outH,
		int	outW,
		int	mode
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, eve::tensor::Resize2d , 以及 x.

emitRMSNorm()

Tensor * eve::tensor::Func::emitRMSNorm	(	const Tensor *	x,
		const Tensor *	scale,
		float	eps
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, eve::tensor::RMSNorm, scale , 以及 x.

emitSdpa()

Tensor * eve::tensor::Func::emitSdpa	(	const Tensor *	q,
		const Tensor *	k,
		const Tensor *	v,
		const Tensor *	mask,
		float	scale
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), n, eve::tensor::Graph::node(), scale, eve::tensor::ScaledDotProductAttention , 以及 v.

emitSlice()

Tensor * eve::tensor::Func::emitSlice	(	const Tensor *	x,
		int	axis,
		int	begin,
		int	end
	)

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::GraphNode::dims, ensureNode(), eve::tensor::Tensor::kMaxRank, n, eve::tensor::Graph::node(), eve::tensor::GraphNode::rank, eve::tensor::Slice , 以及 x.

emitSoftmax()

Tensor * eve::tensor::Func::emitSoftmax	(	const Tensor *	x,
		int	axis,
		bool	logMode
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), eve::tensor::GraphNode::in0, eve::tensor::LogSoftmax, n, eve::tensor::Graph::node(), eve::tensor::Softmax , 以及 x.

emitTernary()

Tensor * eve::tensor::Func::emitTernary	(	OpType	type,
		const Tensor *	a,
		const Tensor *	b,
		const Tensor *	c
	)

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

引用了 a, eve::tensor::Graph::addNode(), b, c, ensureNode(), n, eve::tensor::Graph::node() , 以及 type.

emitTranspose()

Tensor * eve::tensor::Func::emitTranspose

(

const Tensor *

x

)

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

引用了 emitPermute(), ensureNode(), eve::tensor::Graph::node() , 以及 x.

emitUnary()

Tensor * eve::tensor::Func::emitUnary	(	OpType	type,
		const Tensor *	x
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), n, eve::tensor::Graph::node(), type , 以及 x.

emitUnaryScalar()

Tensor * eve::tensor::Func::emitUnaryScalar	(	OpType	type,
		const Tensor *	x,
		float	s0,
		float	s1 = 0.f
	)

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

引用了 eve::tensor::Graph::addNode(), ensureNode(), n, eve::tensor::Graph::node(), type , 以及 x.

ensureNode()

int eve::tensor::Func::ensureNode

(

const Tensor *

t

)

Ensure tensor is a node in this graph (Const-capture if eager).

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

引用了 eve::tensor::Graph::addNode(), eve::tensor::Const, eve::tensor::Tensor::data(), eve::tensor::Tensor::ensureEager(), eve::tensor::Tensor::getSize(), eve::tensor::Tensor::graph(), eve::tensor::Tensor::isQuantized(), eve::tensor::Tensor::isSymbolic(), n, eve::tensor::Tensor::nodeId(), eve::tensor::Tensor::qBytes(), eve::tensor::Tensor::qGroup() , 以及 eve::tensor::Tensor::qScales().

被这些函数引用 emitArgMax(), emitBinary(), emitCast(), emitConcat(), emitConv1d(), emitConv2d(), emitEmbedding(), emitLayerNorm(), emitMatMul(), emitPermute(), emitPool(), emitReduce(), emitReshape(), emitResize2d(), emitRMSNorm(), emitSdpa(), emitSlice(), emitSoftmax(), emitTernary(), emitTranspose(), emitUnary(), emitUnaryScalar() , 以及 setOutput().

graph() [1/2]

Graph & eve::tensor::Func::graph ( )

inline

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

graph() [2/2]

const Graph & eve::tensor::Func::graph ( ) const

inline

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

input1()

Tensor * eve::tensor::Func::input1

(

int

d0

)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

input2()

Tensor * eve::tensor::Func::input2	(	int	d0,
		int	d1
	)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

input3()

Tensor * eve::tensor::Func::input3	(	int	d0,
		int	d1,
		int	d2
	)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

input4()

Tensor * eve::tensor::Func::input4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

input5()

Tensor * eve::tensor::Func::input5	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

input6()

Tensor * eve::tensor::Func::input6	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

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

引用了 eve::tensor::Graph::addNode(), d , 以及 eve::tensor::Placeholder.

isTracing()

bool eve::tensor::Func::isTracing ( ) const

inline

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

outputNode()

int eve::tensor::Func::outputNode ( ) const

inline

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

owner()

TF * eve::tensor::Func::owner ( ) const

inline

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

placeholderCount()

int eve::tensor::Func::placeholderCount ( ) const

inline

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

setOutput()

void eve::tensor::Func::setOutput

(

Tensor *

t

)

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

引用了 ensureNode().

---

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

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