TF2-like namespace module. Script: tf <- eve.TF(); Default eager; tf.func() traces a graph for compile/run. 更多...

#include <TF.h>

类 eve::tensor::TF 继承关系图:

Public 成员函数
	Module_REG (TF)

	TF ()

	~TF () override=default

Func *	func ()

void	pushTrace (Func *f)

void	popTrace (Func *f)

Func *	tracing () const

Tensor *	zeros1 (int d0)

Tensor *	zeros2 (int d0, int d1)

Tensor *	zeros3 (int d0, int d1, int d2)

Tensor *	zeros4 (int d0, int d1, int d2, int d3)

Tensor *	zeros5 (int d0, int d1, int d2, int d3, int d4)

Tensor *	zeros6 (int d0, int d1, int d2, int d3, int d4, int d5)

Tensor *	ones1 (int d0)

Tensor *	ones2 (int d0, int d1)

Tensor *	ones3 (int d0, int d1, int d2)

Tensor *	ones4 (int d0, int d1, int d2, int d3)

Tensor *	ones5 (int d0, int d1, int d2, int d3, int d4)

Tensor *	ones6 (int d0, int d1, int d2, int d3, int d4, int d5)

Tensor *	fill1 (int d0, float value)

Tensor *	fill2 (int d0, int d1, float value)

Tensor *	fill3 (int d0, int d1, int d2, float value)

Tensor *	fill4 (int d0, int d1, int d2, int d3, float value)

Tensor *	constantScalar (float value)

Tensor *	arange (int n)

Tensor *	linspace (float start, float end, int n)

Tensor *	eye (int n)

Tensor *	randomUniform1 (int d0)

Tensor *	randomUniform2 (int d0, int d1)

Tensor *	randomUniform3 (int d0, int d1, int d2)

Tensor *	randomUniform4 (int d0, int d1, int d2, int d3)

Tensor *	randomNormal1 (int d0)

Tensor *	randomNormal2 (int d0, int d1)

Tensor *	randomNormal3 (int d0, int d1, int d2)

Tensor *	randomNormal4 (int d0, int d1, int d2, int d3)

Tensor *	rand1 (int d0)

Tensor *	rand2 (int d0, int d1)

Tensor *	rand3 (int d0, int d1, int d2)

Tensor *	rand4 (int d0, int d1, int d2, int d3)

Tensor *	randn1 (int d0)

Tensor *	randn2 (int d0, int d1)

Tensor *	randn3 (int d0, int d1, int d2)

Tensor *	randn4 (int d0, int d1, int d2, int d3)

void	setRandomSeed (uint32_t seed)

uint32_t	getRandomSeed () const

Tensor *	add (Tensor a, Tensor b)

Tensor *	sub (Tensor a, Tensor b)

Tensor *	multiply (Tensor a, Tensor b)

Tensor *	div (Tensor a, Tensor b)

Tensor *	addScalar (Tensor *a, float s)

Tensor *	subScalar (Tensor *a, float s)

Tensor *	mulScalar (Tensor *a, float s)

Tensor *	divScalar (Tensor *a, float s)

Tensor *	neg (Tensor *a)

Tensor *	abs (Tensor *a)

Tensor *	sqrt (Tensor *a)

Tensor *	exp (Tensor *a)

Tensor *	log (Tensor *a)

Tensor *	sin (Tensor *a)

Tensor *	cos (Tensor *a)

Tensor *	tanh (Tensor *a)

Tensor *	relu (Tensor *a)

Tensor *	sigmoid (Tensor *a)

Tensor *	gelu (Tensor *a)

Tensor *	silu (Tensor *a)

Tensor *	powScalar (Tensor *a, float exp)

Tensor *	clamp (Tensor *a, float lo, float hi)

Tensor *	maximumScalar (Tensor *a, float s)

Tensor *	minimumScalar (Tensor *a, float s)

Tensor *	matmul (Tensor a, Tensor b)

Tensor *	transpose (Tensor *a)

Tensor *	permute2 (Tensor *a, int a0, int a1)

Tensor *	permute3 (Tensor *a, int a0, int a1, int a2)

Tensor *	permute4 (Tensor *a, int a0, int a1, int a2, int a3)

Tensor *	permute5 (Tensor *a, int a0, int a1, int a2, int a3, int a4)

Tensor *	permute6 (Tensor *a, int a0, int a1, int a2, int a3, int a4, int a5)

Tensor *	reshape1 (Tensor *a, int d0)

Tensor *	reshape2 (Tensor *a, int d0, int d1)

Tensor *	reshape3 (Tensor *a, int d0, int d1, int d2)

Tensor *	reshape4 (Tensor *a, int d0, int d1, int d2, int d3)

Tensor *	reshape5 (Tensor *a, int d0, int d1, int d2, int d3, int d4)

Tensor *	reshape6 (Tensor *a, int d0, int d1, int d2, int d3, int d4, int d5)

Tensor *	flatten (Tensor *a)

Tensor *	where (Tensor cond, Tensor a, Tensor *b)

Tensor *	concatN (Tensor const ins, int n, int axis)

Tensor *	softmax (Tensor *a, int axis)

Tensor *	logSoftmax (Tensor *a, int axis)

Tensor *	layernorm (Tensor *a, float eps)

Tensor *	layernormWB (Tensor a, Tensor scale, Tensor *bias, float eps)

Tensor *	rmsnorm (Tensor *a, float eps)

Tensor *	rmsnormW (Tensor a, Tensor scale, float eps)

Tensor *	conv1d (Tensor x, Tensor w, int stride, int pad)

Tensor *	conv1dBias (Tensor x, Tensor w, Tensor *bias, int stride, int pad)

Tensor *	conv2d (Tensor x, Tensor w, int stride, int pad)

Tensor *	conv2dBias (Tensor x, Tensor w, Tensor *bias, int stride, int pad)

Tensor *	maxpool2d (Tensor *x, int ksize, int stride, int pad)

Tensor *	avgpool2d (Tensor *x, int ksize, int stride, int pad)

Tensor *	embedding (Tensor table, Tensor indices)

Tensor *	concat2 (Tensor a, Tensor b, int axis)

Tensor *	concat3 (Tensor a, Tensor b, Tensor *c, int axis)

Tensor *	concat4 (Tensor a, Tensor b, Tensor c, Tensor d, int axis)

Tensor *	slice (Tensor *a, int axis, int begin, int end)

Tensor *	sumAxis (Tensor *a, int axis, int keepDims)

Tensor *	meanAxis (Tensor *a, int axis, int keepDims)

Tensor *	minAxis (Tensor *a, int axis, int keepDims)

Tensor *	maxAxis (Tensor *a, int axis, int keepDims)

Tensor *	argmax (Tensor *a, int axis, int keepDims)

Tensor *	cast (Tensor *a, const std::string &dtype)

Tensor *	sdpa (Tensor q, Tensor k, Tensor *v, float scale)

Tensor *	sdpaMasked (Tensor q, Tensor k, Tensor v, Tensor mask, float scale)

Tensor *	resize2d (Tensor *a, int outW, int outH, int mode)

Tensor *	quantizeWeight (Tensor *a, const std::string &dtype, int group=0)

float	reduceSum (Tensor *a)

float	reduceMean (Tensor *a)

float	reduceMin (Tensor *a)

float	reduceMax (Tensor *a)

Public 成员函数继承自 eve::Module
virtual	~Module ()

virtual std::string	getName () const =0

详细描述

TF2-like namespace module. Script: tf <- eve.TF(); Default eager; tf.func() traces a graph for compile/run.

在文件 TF.h 第 18 行定义.

构造及析构函数说明

◆ TF()

eve::tensor::TF::TF ( )

在文件 TF.cpp 第 52 行定义.

◆ ~TF()

eve::tensor::TF::~TF ( )

overridedefault

成员函数说明

◆ abs()

Tensor * eve::tensor::TF::abs ( Tensor * a )

◆ add()

Tensor * eve::tensor::TF::add	(	Tensor *	a,
		Tensor *	b
	)

◆ addScalar()

Tensor * eve::tensor::TF::addScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 307 行定义.

引用了 a, eve::tensor::AddScalar, f, s , 以及 tracing().

◆ arange()

Tensor * eve::tensor::TF::arange ( int n )

在文件 TF.cpp 第 176 行定义.

引用了 n , 以及 tracing().

◆ argmax()

Tensor * eve::tensor::TF::argmax	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

在文件 TF.cpp 第 733 行定义.

引用了 a, eve::tensor::kernels::argmax(), f, eve::tensor::Int32, eve::tensor::Tensor::kMaxRank , 以及 tracing().

◆ avgpool2d()

Tensor * eve::tensor::TF::avgpool2d	(	Tensor *	x,
		int	ksize,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 585 行定义.

引用了 eve::tensor::kernels::avgpool2d(), eve::tensor::AvgPool2d, f, tracing() , 以及 x.

◆ cast()

Tensor * eve::tensor::TF::cast	(	Tensor *	a,
		const std::string &	dtype
	)

在文件 TF.cpp 第 748 行定义.

引用了 a, f, eve::tensor::Float32, eve::tensor::Tensor::kMaxRank, eve::tensor::parseDType() , 以及 tracing().

◆ clamp()

Tensor * eve::tensor::TF::clamp	(	Tensor *	a,
		float	lo,
		float	hi
	)

在文件 TF.cpp 第 332 行定义.

引用了 a, eve::tensor::Clamp, f , 以及 tracing().

◆ concat2()

Tensor * eve::tensor::TF::concat2	(	Tensor *	a,
		Tensor *	b,
		int	axis
	)

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

引用了 a, b , 以及 concatN().

◆ concat3()

Tensor * eve::tensor::TF::concat3	(	Tensor *	a,
		Tensor *	b,
		Tensor *	c,
		int	axis
	)

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

引用了 a, b, c , 以及 concatN().

◆ concat4()

Tensor * eve::tensor::TF::concat4	(	Tensor *	a,
		Tensor *	b,
		Tensor *	c,
		Tensor *	d,
		int	axis
	)

在文件 TF.cpp 第 643 行定义.

引用了 a, b, c, concatN() , 以及 d.

◆ concatN()

Tensor * eve::tensor::TF::concatN	(	Tensor const	ins,
		int	n,
		int	axis
	)

在文件 TF.cpp 第 684 行定义.

引用了 f, n , 以及 tracing().

被这些函数引用 concat2(), concat3() , 以及 concat4().

◆ constantScalar()

Tensor * eve::tensor::TF::constantScalar ( float value )

在文件 TF.cpp 第 171 行定义.

引用了 d , 以及 value.

◆ conv1d()

Tensor * eve::tensor::TF::conv1d	(	Tensor *	x,
		Tensor *	w,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 534 行定义.

引用了 conv1dBias(), w , 以及 x.

◆ conv1dBias()

Tensor * eve::tensor::TF::conv1dBias	(	Tensor *	x,
		Tensor *	w,
		Tensor *	bias,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 538 行定义.

引用了 eve::tensor::kernels::conv1d(), eve::tensor::Tensor::data(), f, eve::tensor::Tensor::getSize(), tracing(), w , 以及 x.

被这些函数引用 conv1d().

◆ conv2d()

Tensor * eve::tensor::TF::conv2d	(	Tensor *	x,
		Tensor *	w,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 553 行定义.

引用了 conv2dBias(), w , 以及 x.

◆ conv2dBias()

Tensor * eve::tensor::TF::conv2dBias	(	Tensor *	x,
		Tensor *	w,
		Tensor *	bias,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 557 行定义.

引用了 eve::tensor::kernels::conv2d(), eve::tensor::Tensor::data(), f, eve::tensor::Tensor::getSize(), tracing(), w , 以及 x.

被这些函数引用 conv2d().

◆ cos()

Tensor * eve::tensor::TF::cos ( Tensor * a )

◆ div()

Tensor * eve::tensor::TF::div	(	Tensor *	a,
		Tensor *	b
	)

◆ divScalar()

Tensor * eve::tensor::TF::divScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 322 行定义.

引用了 a, eve::tensor::DivScalar, f, s , 以及 tracing().

◆ embedding()

Tensor * eve::tensor::TF::embedding	(	Tensor *	table,
		Tensor *	indices
	)

在文件 TF.cpp 第 597 行定义.

引用了 eve::tensor::Tensor::data(), eve::tensor::Tensor::dequantized(), eve::tensor::kernels::embedding(), f, eve::tensor::Tensor::getDim(), eve::tensor::Tensor::getRank(), eve::tensor::Tensor::getSize(), eve::tensor::Tensor::isQuantized(), eve::tensor::Tensor::kMaxRank , 以及 tracing().

◆ exp()

Tensor * eve::tensor::TF::exp ( Tensor * a )

被这些函数引用 powScalar().

◆ eye()

Tensor * eve::tensor::TF::eye ( int n )

在文件 TF.cpp 第 197 行定义.

引用了 f, eve::tensor::Tensor::makeSymbolic(), n , 以及 tracing().

◆ fill1()

Tensor * eve::tensor::TF::fill1	(	int	d0,
		float	value
	)

在文件 TF.cpp 第 154 行定义.

引用了 d , 以及 value.

◆ fill2()

Tensor * eve::tensor::TF::fill2	(	int	d0,
		int	d1,
		float	value
	)

在文件 TF.cpp 第 158 行定义.

引用了 d , 以及 value.

◆ fill3()

Tensor * eve::tensor::TF::fill3	(	int	d0,
		int	d1,
		int	d2,
		float	value
	)

在文件 TF.cpp 第 162 行定义.

引用了 d , 以及 value.

◆ fill4()

Tensor * eve::tensor::TF::fill4	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		float	value
	)

在文件 TF.cpp 第 166 行定义.

引用了 d , 以及 value.

◆ flatten()

Tensor * eve::tensor::TF::flatten ( Tensor * a )

在文件 TF.cpp 第 428 行定义.

引用了 a, d, f , 以及 tracing().

◆ func()

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

在文件 TF.cpp 第 95 行定义.

◆ gelu()

Tensor * eve::tensor::TF::gelu ( Tensor * a )

◆ getRandomSeed()

uint32_t eve::tensor::TF::getRandomSeed ( ) const

在文件 TF.cpp 第 71 行定义.

◆ layernorm()

Tensor * eve::tensor::TF::layernorm	(	Tensor *	a,
		float	eps
	)

在文件 TF.cpp 第 479 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::layernorm() , 以及 tracing().

◆ layernormWB()

Tensor * eve::tensor::TF::layernormWB	(	Tensor *	a,
		Tensor *	scale,
		Tensor *	bias,
		float	eps
	)

在文件 TF.cpp 第 492 行定义.

引用了 a, eve::tensor::Tensor::data(), f, eve::tensor::Tensor::getSize(), eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::layernorm(), scale , 以及 tracing().

◆ linspace()

Tensor * eve::tensor::TF::linspace	(	float	start,
		float	end,
		int	n
	)

在文件 TF.cpp 第 184 行定义.

引用了 n, step , 以及 tracing().

◆ log()

Tensor * eve::tensor::TF::log ( Tensor * a )

◆ logSoftmax()

Tensor * eve::tensor::TF::logSoftmax	(	Tensor *	a,
		int	axis
	)

在文件 TF.cpp 第 467 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::softmax() , 以及 tracing().

◆ matmul()

Tensor * eve::tensor::TF::matmul	(	Tensor *	a,
		Tensor *	b
	)

在文件 TF.cpp 第 348 行定义.

引用了 a, b, f , 以及 tracing().

◆ maxAxis()

Tensor * eve::tensor::TF::maxAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

◆ maximumScalar()

Tensor * eve::tensor::TF::maximumScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 337 行定义.

引用了 a, f, eve::tensor::MaximumScalar, s , 以及 tracing().

◆ maxpool2d()

Tensor * eve::tensor::TF::maxpool2d	(	Tensor *	x,
		int	ksize,
		int	stride,
		int	pad
	)

在文件 TF.cpp 第 573 行定义.

引用了 f, eve::tensor::kernels::maxpool2d(), eve::tensor::MaxPool2d, tracing() , 以及 x.

◆ meanAxis()

Tensor * eve::tensor::TF::meanAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

◆ minAxis()

Tensor * eve::tensor::TF::minAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

◆ minimumScalar()

Tensor * eve::tensor::TF::minimumScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 342 行定义.

引用了 a, f, eve::tensor::MinimumScalar, s , 以及 tracing().

◆ Module_REG()

eve::tensor::TF::Module_REG ( TF )

◆ mulScalar()

Tensor * eve::tensor::TF::mulScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 317 行定义.

引用了 a, f, eve::tensor::MulScalar, s , 以及 tracing().

◆ multiply()

Tensor * eve::tensor::TF::multiply	(	Tensor *	a,
		Tensor *	b
	)

◆ neg()

Tensor * eve::tensor::TF::neg ( Tensor * a )

◆ ones1()

Tensor * eve::tensor::TF::ones1 ( int d0 )

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

引用了 d.

◆ ones2()

Tensor * eve::tensor::TF::ones2	(	int	d0,
		int	d1
	)

在文件 TF.cpp 第 133 行定义.

引用了 d.

◆ ones3()

Tensor * eve::tensor::TF::ones3	(	int	d0,
		int	d1,
		int	d2
	)

在文件 TF.cpp 第 137 行定义.

引用了 d.

◆ ones4()

Tensor * eve::tensor::TF::ones4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

在文件 TF.cpp 第 141 行定义.

引用了 d.

◆ ones5()

Tensor * eve::tensor::TF::ones5	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

在文件 TF.cpp 第 145 行定义.

引用了 d.

◆ ones6()

Tensor * eve::tensor::TF::ones6	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

在文件 TF.cpp 第 149 行定义.

引用了 d.

◆ permute2()

Tensor * eve::tensor::TF::permute2	(	Tensor *	a,
		int	a0,
		int	a1
	)

在文件 TF.cpp 第 360 行定义.

引用了 a, f , 以及 tracing().

◆ permute3()

Tensor * eve::tensor::TF::permute3	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2
	)

在文件 TF.cpp 第 366 行定义.

引用了 a, f , 以及 tracing().

◆ permute4()

Tensor * eve::tensor::TF::permute4	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3
	)

在文件 TF.cpp 第 372 行定义.

引用了 a, f , 以及 tracing().

◆ permute5()

Tensor * eve::tensor::TF::permute5	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3,
		int	a4
	)

在文件 TF.cpp 第 378 行定义.

引用了 a, f , 以及 tracing().

◆ permute6()

Tensor * eve::tensor::TF::permute6	(	Tensor *	a,
		int	a0,
		int	a1,
		int	a2,
		int	a3,
		int	a4,
		int	a5
	)

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

引用了 a, f , 以及 tracing().

◆ popTrace()

void eve::tensor::TF::popTrace ( Func * f )

在文件 TF.cpp 第 77 行定义.

引用了 f.

被这些函数引用 eve::tensor::Func::compile() , 以及 eve::tensor::Func::~Func().

◆ powScalar()

Tensor * eve::tensor::TF::powScalar	(	Tensor *	a,
		float	exp
	)

在文件 TF.cpp 第 327 行定义.

引用了 a, exp(), f, eve::tensor::PowScalar , 以及 tracing().

◆ pushTrace()

void eve::tensor::TF::pushTrace ( Func * f )

在文件 TF.cpp 第 73 行定义.

引用了 f.

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

◆ quantizeWeight()

Tensor * eve::tensor::TF::quantizeWeight	(	Tensor *	a,
		const std::string &	dtype,
		int	group = `0`
	)

Weight-only quantization (eager): pack a float32 tensor into one of the packed weight dtypes — "fp16", "fp8", "fp4", "int8", "int4". int8/int4 use symmetric per-group scales (group = elements per scale, default all).

在文件 TF.cpp 第 616 行定义.

引用了 a, eve::tensor::Float32, group, eve::tensor::q::isQuantDType(), p, eve::tensor::parseDType(), eve::tensor::q::quantize() , 以及 tracing().

◆ rand1()

Tensor * eve::tensor::TF::rand1 ( int d0 )

inline

在文件 TF.h 第 65 行定义.

引用了 randomUniform1().

◆ rand2()

Tensor * eve::tensor::TF::rand2	(	int	d0,
		int	d1
	)

inline

在文件 TF.h 第 66 行定义.

引用了 randomUniform2().

◆ rand3()

Tensor * eve::tensor::TF::rand3	(	int	d0,
		int	d1,
		int	d2
	)

inline

在文件 TF.h 第 67 行定义.

引用了 randomUniform3().

◆ rand4()

Tensor * eve::tensor::TF::rand4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

inline

在文件 TF.h 第 68 行定义.

引用了 randomUniform4().

◆ randn1()

Tensor * eve::tensor::TF::randn1 ( int d0 )

inline

在文件 TF.h 第 71 行定义.

引用了 randomNormal1().

◆ randn2()

Tensor * eve::tensor::TF::randn2	(	int	d0,
		int	d1
	)

inline

在文件 TF.h 第 72 行定义.

引用了 randomNormal2().

◆ randn3()

Tensor * eve::tensor::TF::randn3	(	int	d0,
		int	d1,
		int	d2
	)

inline

在文件 TF.h 第 73 行定义.

引用了 randomNormal3().

◆ randn4()

Tensor * eve::tensor::TF::randn4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

inline

在文件 TF.h 第 74 行定义.

引用了 randomNormal4().

◆ randomNormal1()

Tensor * eve::tensor::TF::randomNormal1 ( int d0 )

在文件 TF.cpp 第 240 行定义.

引用了 tracing() , 以及 zeros1().

被这些函数引用 randn1().

◆ randomNormal2()

Tensor * eve::tensor::TF::randomNormal2	(	int	d0,
		int	d1
	)

在文件 TF.cpp 第 247 行定义.

引用了 tracing() , 以及 zeros2().

被这些函数引用 randn2().

◆ randomNormal3()

Tensor * eve::tensor::TF::randomNormal3	(	int	d0,
		int	d1,
		int	d2
	)

在文件 TF.cpp 第 254 行定义.

引用了 tracing() , 以及 zeros3().

被这些函数引用 randn3().

◆ randomNormal4()

Tensor * eve::tensor::TF::randomNormal4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

在文件 TF.cpp 第 261 行定义.

引用了 tracing() , 以及 zeros4().

被这些函数引用 randn4().

◆ randomUniform1()

Tensor * eve::tensor::TF::randomUniform1 ( int d0 )

在文件 TF.cpp 第 212 行定义.

引用了 tracing() , 以及 zeros1().

被这些函数引用 rand1().

◆ randomUniform2()

Tensor * eve::tensor::TF::randomUniform2	(	int	d0,
		int	d1
	)

在文件 TF.cpp 第 219 行定义.

引用了 tracing() , 以及 zeros2().

被这些函数引用 rand2().

◆ randomUniform3()

Tensor * eve::tensor::TF::randomUniform3	(	int	d0,
		int	d1,
		int	d2
	)

在文件 TF.cpp 第 226 行定义.

引用了 tracing() , 以及 zeros3().

被这些函数引用 rand3().

◆ randomUniform4()

Tensor * eve::tensor::TF::randomUniform4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

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

引用了 tracing() , 以及 zeros4().

被这些函数引用 rand4().

◆ reduceMax()

float eve::tensor::TF::reduceMax ( Tensor * a )

在文件 TF.cpp 第 820 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

◆ reduceMean()

float eve::tensor::TF::reduceMean ( Tensor * a )

在文件 TF.cpp 第 804 行定义.

引用了 a, reduceSum() , 以及 tracing().

◆ reduceMin()

float eve::tensor::TF::reduceMin ( Tensor * a )

在文件 TF.cpp 第 811 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

◆ reduceSum()

float eve::tensor::TF::reduceSum ( Tensor * a )

在文件 TF.cpp 第 795 行定义.

引用了 a, eve::tensor::gpuReduce() , 以及 tracing().

被这些函数引用 reduceMean().

◆ relu()

Tensor * eve::tensor::TF::relu ( Tensor * a )

◆ reshape1()

Tensor * eve::tensor::TF::reshape1	(	Tensor *	a,
		int	d0
	)

在文件 TF.cpp 第 391 行定义.

引用了 a, d, f , 以及 tracing().

◆ reshape2()

Tensor * eve::tensor::TF::reshape2	(	Tensor *	a,
		int	d0,
		int	d1
	)

在文件 TF.cpp 第 397 行定义.

引用了 a, d, f , 以及 tracing().

◆ reshape3()

Tensor * eve::tensor::TF::reshape3	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2
	)

在文件 TF.cpp 第 403 行定义.

引用了 a, d, f , 以及 tracing().

◆ reshape4()

Tensor * eve::tensor::TF::reshape4	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3
	)

在文件 TF.cpp 第 409 行定义.

引用了 a, d, f , 以及 tracing().

◆ reshape5()

Tensor * eve::tensor::TF::reshape5	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

在文件 TF.cpp 第 415 行定义.

引用了 a, d, f , 以及 tracing().

◆ reshape6()

Tensor * eve::tensor::TF::reshape6	(	Tensor *	a,
		int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

在文件 TF.cpp 第 421 行定义.

引用了 a, d, f , 以及 tracing().

◆ resize2d()

Tensor * eve::tensor::TF::resize2d	(	Tensor *	a,
		int	outW,
		int	outH,
		int	mode
	)

在文件 TF.cpp 第 784 行定义.

引用了 a, f, eve::tensor::kernels::resize2d() , 以及 tracing().

◆ rmsnorm()

Tensor * eve::tensor::TF::rmsnorm	(	Tensor *	a,
		float	eps
	)

在文件 TF.cpp 第 507 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::rmsnorm() , 以及 tracing().

◆ rmsnormW()

Tensor * eve::tensor::TF::rmsnormW	(	Tensor *	a,
		Tensor *	scale,
		float	eps
	)

在文件 TF.cpp 第 520 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::rmsnorm(), scale , 以及 tracing().

◆ sdpa()

Tensor * eve::tensor::TF::sdpa	(	Tensor *	q,
		Tensor *	k,
		Tensor *	v,
		float	scale
	)

在文件 TF.cpp 第 761 行定义.

引用了 scale, sdpaMasked() , 以及 v.

◆ sdpaMasked()

Tensor * eve::tensor::TF::sdpaMasked	(	Tensor *	q,
		Tensor *	k,
		Tensor *	v,
		Tensor *	mask,
		float	scale
	)

在文件 TF.cpp 第 765 行定义.

引用了 eve::tensor::Tensor::data(), f, eve::tensor::Tensor::getDim(), eve::tensor::Tensor::getRank(), scale, eve::tensor::kernels::sdpa(), tracing() , 以及 v.

被这些函数引用 sdpa().

◆ setRandomSeed()

void eve::tensor::TF::setRandomSeed ( uint32_t seed )

在文件 TF.cpp 第 66 行定义.

引用了 seed.

◆ sigmoid()

Tensor * eve::tensor::TF::sigmoid ( Tensor * a )

◆ silu()

Tensor * eve::tensor::TF::silu ( Tensor * a )

◆ sin()

Tensor * eve::tensor::TF::sin ( Tensor * a )

◆ slice()

Tensor * eve::tensor::TF::slice	(	Tensor *	a,
		int	axis,
		int	begin,
		int	end
	)

在文件 TF.cpp 第 689 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::sliceOp() , 以及 tracing().

◆ softmax()

Tensor * eve::tensor::TF::softmax	(	Tensor *	a,
		int	axis
	)

在文件 TF.cpp 第 455 行定义.

引用了 a, f, eve::tensor::Tensor::kMaxRank, eve::tensor::kernels::softmax() , 以及 tracing().

◆ sqrt()

Tensor * eve::tensor::TF::sqrt ( Tensor * a )

◆ sub()

Tensor * eve::tensor::TF::sub	(	Tensor *	a,
		Tensor *	b
	)

◆ subScalar()

Tensor * eve::tensor::TF::subScalar	(	Tensor *	a,
		float	s
	)

在文件 TF.cpp 第 312 行定义.

引用了 a, f, s, eve::tensor::SubScalar , 以及 tracing().

◆ sumAxis()

Tensor * eve::tensor::TF::sumAxis	(	Tensor *	a,
		int	axis,
		int	keepDims
	)

◆ tanh()

Tensor * eve::tensor::TF::tanh ( Tensor * a )

◆ tracing()

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

在文件 TF.cpp 第 91 行定义.

被这些函数引用 addScalar(), arange(), argmax(), avgpool2d(), cast(), clamp(), concatN(), conv1dBias(), conv2dBias(), divScalar(), embedding(), eye(), flatten(), layernorm(), layernormWB(), linspace(), logSoftmax(), matmul(), maximumScalar(), maxpool2d(), minimumScalar(), mulScalar(), permute2(), permute3(), permute4(), permute5(), permute6(), powScalar(), quantizeWeight(), randomNormal1(), randomNormal2(), randomNormal3(), randomNormal4(), randomUniform1(), randomUniform2(), randomUniform3(), randomUniform4(), reduceMax(), reduceMean(), reduceMin(), reduceSum(), reshape1(), reshape2(), reshape3(), reshape4(), reshape5(), reshape6(), resize2d(), rmsnorm(), rmsnormW(), sdpaMasked(), slice(), softmax(), subScalar(), transpose() , 以及 where().

◆ transpose()

Tensor * eve::tensor::TF::transpose ( Tensor * a )

在文件 TF.cpp 第 354 行定义.

引用了 a, f , 以及 tracing().

◆ where()

Tensor * eve::tensor::TF::where	(	Tensor *	cond,
		Tensor *	a,
		Tensor *	b
	)

在文件 TF.cpp 第 435 行定义.

引用了 a, b, f, eve::tensor::Tensor::get(), eve::tensor::Tensor::getDim(), eve::tensor::Tensor::getRank(), eve::tensor::Tensor::getSize(), eve::tensor::Tensor::kMaxRank, tracing() , 以及 eve::tensor::Where.

◆ zeros1()

Tensor * eve::tensor::TF::zeros1 ( int d0 )

在文件 TF.cpp 第 104 行定义.

引用了 d.

被这些函数引用 randomNormal1() , 以及 randomUniform1().

◆ zeros2()

Tensor * eve::tensor::TF::zeros2	(	int	d0,
		int	d1
	)

在文件 TF.cpp 第 108 行定义.

引用了 d.

被这些函数引用 randomNormal2() , 以及 randomUniform2().

◆ zeros3()

Tensor * eve::tensor::TF::zeros3	(	int	d0,
		int	d1,
		int	d2
	)

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

引用了 d.

被这些函数引用 randomNormal3() , 以及 randomUniform3().

◆ zeros4()

Tensor * eve::tensor::TF::zeros4	(	int	d0,
		int	d1,
		int	d2,
		int	d3
	)

在文件 TF.cpp 第 116 行定义.

引用了 d.

被这些函数引用 randomNormal4() , 以及 randomUniform4().

◆ zeros5()

Tensor * eve::tensor::TF::zeros5	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4
	)

在文件 TF.cpp 第 120 行定义.

引用了 d.

◆ zeros6()

Tensor * eve::tensor::TF::zeros6	(	int	d0,
		int	d1,
		int	d2,
		int	d3,
		int	d4,
		int	d5
	)

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

引用了 d.

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

src/modules/tensor/TF.h
src/modules/tensor/TF.cpp

Public 成员函数

详细描述

构造及析构函数说明

◆ TF()

◆ ~TF()

成员函数说明

◆ abs()

◆ add()

◆ addScalar()

◆ arange()

◆ argmax()

◆ avgpool2d()

◆ cast()

◆ clamp()

◆ concat2()

◆ concat3()

◆ concat4()

◆ concatN()

◆ constantScalar()

◆ conv1d()

◆ conv1dBias()

◆ conv2d()

◆ conv2dBias()

◆ cos()

◆ div()

◆ divScalar()

◆ embedding()

◆ exp()

◆ eye()

◆ fill1()

◆ fill2()

◆ fill3()

◆ fill4()

◆ flatten()

◆ func()

◆ gelu()

◆ getRandomSeed()

◆ layernorm()

◆ layernormWB()

◆ linspace()

◆ log()

◆ logSoftmax()

◆ matmul()

◆ maxAxis()

◆ maximumScalar()

◆ maxpool2d()

◆ meanAxis()

◆ minAxis()

◆ minimumScalar()

◆ Module_REG()

◆ mulScalar()

◆ multiply()

◆ neg()

◆ ones1()

◆ ones2()

◆ ones3()

◆ ones4()

◆ ones5()

◆ ones6()

◆ permute2()

◆ permute3()

◆ permute4()

◆ permute5()

◆ permute6()

◆ popTrace()

◆ powScalar()

◆ pushTrace()

◆ quantizeWeight()

◆ rand1()

◆ rand2()

◆ rand3()

◆ rand4()

◆ randn1()

◆ randn2()

◆ randn3()

◆ randn4()

◆ randomNormal1()

◆ randomNormal2()

◆ randomNormal3()

◆ randomNormal4()