eve::tensor::GraphNode结构体 参考

GraphNode public API. 更多...

#include <Graph.h>

Public 属性
OpType	type = OpType::Const
int	dims [Tensor::kMaxRank] = {0, 0, 0, 0, 0, 0}
int	rank = 0
int	size = 0
int	in0 = -1
int	in1 = -1
int	in2 = -1
int	in3 = -1
int	in4 = -1
float	s0 = 0.f
float	s1 = 0.f
float	s2 = 0.f
float	s3 = 0.f
int	i0 = 0
int	i1 = 0
int	i2 = 0
int	i3 = 0
int	perm [Tensor::kMaxRank] = {0, 1, 2, 3, 4, 5}
int	permRank = 0
int	placeholderSlot = -1
int	dtype = static_cast<int>(DType::Float32)
std::vector< float >	constData
std::vector< uint8_t >	constBytes
std::vector< float >	constScales
int	qGroup = 0

详细描述

GraphNode public API.

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

类成员变量说明

constBytes

std::vector<uint8_t> eve::tensor::GraphNode::constBytes

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

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding() , 以及 eve::tensor::GpuProgram::tryBuild().

constData

std::vector<float> eve::tensor::GraphNode::constData

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

constScales

std::vector<float> eve::tensor::GraphNode::constScales

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

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

dims

int eve::tensor::GraphNode::dims[Tensor::kMaxRank] = {0, 0, 0, 0, 0, 0}

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

被这些函数引用 eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), eve::tensor::glsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::emitInputIndexExprs(), eve::tensor::Func::emitLayerNorm(), eve::tensor::Func::emitPool(), eve::tensor::Func::emitReduce(), eve::tensor::Func::emitResize2d(), eve::tensor::Func::emitRMSNorm(), eve::tensor::Func::emitSlice(), eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice() , 以及 eve::tensor::glsl_detail::genSlice().

dtype

int eve::tensor::GraphNode::dtype = static_cast<int>(DType::Float32)

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

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding() , 以及 eve::tensor::glsl_detail::genEmbedding().

i0

int eve::tensor::GraphNode::i0 = 0

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

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax() , 以及 eve::tensor::glsl_detail::genSoftmax().

i1

int eve::tensor::GraphNode::i1 = 0

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

被这些函数引用 eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genSlice() , 以及 eve::tensor::glsl_detail::genSlice().

i2

int eve::tensor::GraphNode::i2 = 0

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

被这些函数引用 eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genSlice() , 以及 eve::tensor::glsl_detail::genSlice().

i3

int eve::tensor::GraphNode::i3 = 0

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

in0

int eve::tensor::GraphNode::in0 = -1

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::ChainContext::emit(), eve::tensor::Func::emitSoftmax(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::wgsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genReduceOrArgmax(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSdpa(), eve::tensor::glsl_detail::genSdpa(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax() , 以及 eve::tensor::glsl_detail::genSoftmax().

in1

int eve::tensor::GraphNode::in1 = -1

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::ChainContext::emit(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genSdpa() , 以及 eve::tensor::glsl_detail::genSdpa().

in2

int eve::tensor::GraphNode::in2 = -1

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::ChainContext::emit(), eve::tensor::glsl_detail::genConv() , 以及 eve::tensor::wgsl_detail::genConv().

in3

int eve::tensor::GraphNode::in3 = -1

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

in4

int eve::tensor::GraphNode::in4 = -1

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

perm

int eve::tensor::GraphNode::perm[Tensor::kMaxRank] = {0, 1, 2, 3, 4, 5}

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

被这些函数引用 eve::tensor::wgsl_detail::genPermute() , 以及 eve::tensor::glsl_detail::genPermute().

permRank

int eve::tensor::GraphNode::permRank = 0

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

placeholderSlot

int eve::tensor::GraphNode::placeholderSlot = -1

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

qGroup

int eve::tensor::GraphNode::qGroup = 0

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

被这些函数引用 eve::tensor::wgsl_detail::genEmbedding() , 以及 eve::tensor::glsl_detail::genEmbedding().

rank

int eve::tensor::GraphNode::rank = 0

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

被这些函数引用 eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), eve::tensor::glsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::emitInputIndexExprs(), eve::tensor::Func::emitLayerNorm(), eve::tensor::Func::emitPool(), eve::tensor::Func::emitReduce(), eve::tensor::Func::emitResize2d(), eve::tensor::Func::emitRMSNorm(), eve::tensor::Func::emitSlice(), eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice() , 以及 eve::tensor::GpuProgram::tryBuild().

s0

float eve::tensor::GraphNode::s0 = 0.f

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genSdpa() , 以及 eve::tensor::glsl_detail::genSdpa().

s1

float eve::tensor::GraphNode::s1 = 0.f

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit() , 以及 eve::tensor::wgsl_detail::ChainContext::emit().

s2

float eve::tensor::GraphNode::s2 = 0.f

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

s3

float eve::tensor::GraphNode::s3 = 0.f

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

size

int eve::tensor::GraphNode::size = 0

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

被这些函数引用 eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::wgsl_detail::genEmbedding(), eve::tensor::glsl_detail::genEmbedding(), eve::tensor::wgsl_detail::genNorm(), eve::tensor::glsl_detail::genNorm(), eve::tensor::wgsl_detail::genPermute(), eve::tensor::glsl_detail::genPermute(), eve::tensor::glsl_detail::genPool(), eve::tensor::wgsl_detail::genPool(), eve::tensor::glsl_detail::genResize2d(), eve::tensor::wgsl_detail::genResize2d(), eve::tensor::wgsl_detail::genSlice(), eve::tensor::glsl_detail::genSlice(), eve::tensor::wgsl_detail::genSoftmax() , 以及 eve::tensor::glsl_detail::genSoftmax().

type

OpType eve::tensor::GraphNode::type = OpType::Const

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

被这些函数引用 eve::tensor::glsl_detail::ChainContext::emit(), eve::tensor::wgsl_detail::ChainContext::emit(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genPool() , 以及 eve::tensor::wgsl_detail::genPool().

---

该结构体的文档由以下文件生成:

• src/modules/tensor/Graph.h