#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 |
详细描述
类成员变量说明
◆ constBytes
| std::vector<uint8_t> eve::tensor::GraphNode::constBytes |
被这些函数引用 eve::tensor::GpuProgram::tryBuild().
◆ constData
◆ constScales
| std::vector<float> eve::tensor::GraphNode::constScales |
被这些函数引用 eve::tensor::GpuProgram::tryBuild().
◆ dims
| int eve::tensor::GraphNode::dims[Tensor::kMaxRank] = {0, 0, 0, 0, 0, 0} |
被这些函数引用 eve::tensor::Graph::addNode(), eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), eve::tensor::Func::emitLayerNorm(), eve::tensor::Func::emitPool(), eve::tensor::Func::emitReduce(), eve::tensor::Func::emitResize2d(), eve::tensor::Func::emitRMSNorm() , 以及 eve::tensor::Func::emitSlice().
◆ dtype
| int eve::tensor::GraphNode::dtype = static_cast<int>(DType::Float32) |
◆ i0
◆ i1
◆ i2
◆ i3
◆ in0
| int eve::tensor::GraphNode::in0 = -1 |
被这些函数引用 eve::tensor::Func::emitSoftmax() , 以及 eve::tensor::optimizeGraph().
◆ in1
| int eve::tensor::GraphNode::in1 = -1 |
被这些函数引用 eve::tensor::optimizeGraph().
◆ in2
◆ in3
◆ in4
◆ perm
| int eve::tensor::GraphNode::perm[Tensor::kMaxRank] = {0, 1, 2, 3, 4, 5} |
◆ permRank
◆ placeholderSlot
◆ qGroup
◆ rank
| int eve::tensor::GraphNode::rank = 0 |
被这些函数引用 eve::tensor::Graph::addNode(), eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), 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::GpuProgram::tryBuild().
◆ s0
◆ s1
◆ s2
◆ s3
◆ size
| int eve::tensor::GraphNode::size = 0 |
被这些函数引用 eve::tensor::Graph::addNode() , 以及 eve::tensor::optimizeGraph().
◆ type
| OpType eve::tensor::GraphNode::type = OpType::Const |
被这些函数引用 eve::tensor::optimizeGraph().
该结构体的文档由以下文件生成:
- src/modules/tensor/Graph.h