载入中...
搜索中...
未找到
Public 属性 | 所有成员列表
eve::tensor::FusedGroup结构体 参考

FusedGroup public API. 更多...

#include <Optimizer.h>

Public 属性

GroupKind kind = GroupKind::Elementwise
 
OpType op = OpType::Const
 
std::vector< int > nodes
 
std::vector< int > inputs
 
int outputNode = -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 dtype = static_cast<int>(DType::Float32)
 
int biasNode = -1
 
std::vector< int > epilogue
 
bool hasScale = false
 
bool hasBias = false
 
bool masked = false
 
bool logMode = false
 

详细描述

FusedGroup public API.

在文件 Optimizer.h 第 46 行定义.

类成员变量说明

◆ biasNode

int eve::tensor::FusedGroup::biasNode = -1

在文件 Optimizer.h 第 64 行定义.

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

◆ dtype

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

在文件 Optimizer.h 第 61 行定义.

◆ epilogue

std::vector<int> eve::tensor::FusedGroup::epilogue

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

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

◆ hasBias

bool eve::tensor::FusedGroup::hasBias = false

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

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

◆ hasScale

bool eve::tensor::FusedGroup::hasScale = false

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

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

◆ i0

int eve::tensor::FusedGroup::i0 = 0

在文件 Optimizer.h 第 58 行定义.

◆ i1

int eve::tensor::FusedGroup::i1 = 0

在文件 Optimizer.h 第 58 行定义.

◆ i2

int eve::tensor::FusedGroup::i2 = 0

在文件 Optimizer.h 第 58 行定义.

◆ i3

int eve::tensor::FusedGroup::i3 = 0

在文件 Optimizer.h 第 58 行定义.

◆ inputs

std::vector<int> eve::tensor::FusedGroup::inputs

External producer node ids (deduped, ordered by first use).

在文件 Optimizer.h 第 53 行定义.

被这些函数引用 eve::tensor::glsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul(), eve::tensor::glsl_detail::ChainContext::operand(), eve::tensor::wgsl_detail::ChainContext::operand() , 以及 eve::tensor::optimizeGraph().

◆ kind

GroupKind eve::tensor::FusedGroup::kind = GroupKind::Elementwise

在文件 Optimizer.h 第 47 行定义.

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

◆ logMode

bool eve::tensor::FusedGroup::logMode = false

在文件 Optimizer.h 第 69 行定义.

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

◆ masked

bool eve::tensor::FusedGroup::masked = false

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

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

◆ nodes

std::vector<int> eve::tensor::FusedGroup::nodes

Graph node ids in execution order (elementwise chains: all chain nodes).

在文件 Optimizer.h 第 51 行定义.

被这些函数引用 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::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), eve::tensor::glsl_detail::genMatMul(), eve::tensor::wgsl_detail::genMatMul() , 以及 eve::tensor::optimizeGraph().

◆ op

OpType eve::tensor::FusedGroup::op = OpType::Const

Actual op for polymorphic groups (Reduce sum/mean/min/max, Softmax/LogSoftmax).

在文件 Optimizer.h 第 49 行定义.

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

◆ outputNode

int eve::tensor::FusedGroup::outputNode = -1

在文件 Optimizer.h 第 54 行定义.

被这些函数引用 eve::tensor::glsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::emitInputIndexExprs(), eve::tensor::wgsl_detail::genConcat(), eve::tensor::glsl_detail::genConcat(), eve::tensor::glsl_detail::genConv(), eve::tensor::wgsl_detail::genConv(), eve::tensor::glsl_detail::genElementwise(), eve::tensor::wgsl_detail::genElementwise(), 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() , 以及 eve::tensor::optimizeGraph().

◆ perm

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

在文件 Optimizer.h 第 59 行定义.

◆ permRank

int eve::tensor::FusedGroup::permRank = 0

在文件 Optimizer.h 第 60 行定义.

◆ s0

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

在文件 Optimizer.h 第 57 行定义.

◆ s1

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

在文件 Optimizer.h 第 57 行定义.

◆ s2

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

在文件 Optimizer.h 第 57 行定义.

◆ s3

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

在文件 Optimizer.h 第 57 行定义.

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