#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 |
详细描述
在文件 Optimizer.h 第 42 行定义.
类成员变量说明
◆ biasNode
| int eve::tensor::FusedGroup::biasNode = -1 |
在文件 Optimizer.h 第 60 行定义.
被这些函数引用 eve::tensor::optimizeGraph().
◆ dtype
| int eve::tensor::FusedGroup::dtype = static_cast<int>(DType::Float32) |
在文件 Optimizer.h 第 57 行定义.
◆ epilogue
| std::vector<int> eve::tensor::FusedGroup::epilogue |
在文件 Optimizer.h 第 61 行定义.
被这些函数引用 eve::tensor::optimizeGraph().
◆ hasBias
| bool eve::tensor::FusedGroup::hasBias = false |
在文件 Optimizer.h 第 63 行定义.
◆ hasScale
| bool eve::tensor::FusedGroup::hasScale = false |
在文件 Optimizer.h 第 62 行定义.
◆ i0
| int eve::tensor::FusedGroup::i0 = 0 |
在文件 Optimizer.h 第 54 行定义.
◆ i1
| int eve::tensor::FusedGroup::i1 = 0 |
在文件 Optimizer.h 第 54 行定义.
◆ i2
| int eve::tensor::FusedGroup::i2 = 0 |
在文件 Optimizer.h 第 54 行定义.
◆ i3
| int eve::tensor::FusedGroup::i3 = 0 |
在文件 Optimizer.h 第 54 行定义.
◆ inputs
| std::vector<int> eve::tensor::FusedGroup::inputs |
External producer node ids (deduped, ordered by first use).
在文件 Optimizer.h 第 49 行定义.
被这些函数引用 eve::tensor::optimizeGraph().
◆ kind
| GroupKind eve::tensor::FusedGroup::kind = GroupKind::Elementwise |
在文件 Optimizer.h 第 43 行定义.
被这些函数引用 eve::tensor::generateKernel(), eve::tensor::generateMatMulVariant() , 以及 eve::tensor::optimizeGraph().
◆ logMode
| bool eve::tensor::FusedGroup::logMode = false |
在文件 Optimizer.h 第 65 行定义.
◆ masked
| bool eve::tensor::FusedGroup::masked = false |
在文件 Optimizer.h 第 64 行定义.
◆ nodes
| std::vector<int> eve::tensor::FusedGroup::nodes |
Graph node ids in execution order (elementwise chains: all chain nodes).
在文件 Optimizer.h 第 47 行定义.
被这些函数引用 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 第 45 行定义.
◆ outputNode
| int eve::tensor::FusedGroup::outputNode = -1 |
在文件 Optimizer.h 第 50 行定义.
被这些函数引用 eve::tensor::optimizeGraph().
◆ perm
| int eve::tensor::FusedGroup::perm[Tensor::kMaxRank] = {0, 1, 2, 3, 4, 5} |
在文件 Optimizer.h 第 55 行定义.
◆ permRank
| int eve::tensor::FusedGroup::permRank = 0 |
在文件 Optimizer.h 第 56 行定义.
◆ s0
| float eve::tensor::FusedGroup::s0 = 0.f |
在文件 Optimizer.h 第 53 行定义.
◆ s1
| float eve::tensor::FusedGroup::s1 = 0.f |
在文件 Optimizer.h 第 53 行定义.
◆ s2
| float eve::tensor::FusedGroup::s2 = 0.f |
在文件 Optimizer.h 第 53 行定义.
◆ s3
| float eve::tensor::FusedGroup::s3 = 0.f |
在文件 Optimizer.h 第 53 行定义.
该结构体的文档由以下文件生成:
- src/modules/tensor/Optimizer.h