载入中...
搜索中...
未找到
Public 成员函数 | 静态 Public 成员函数 | 所有成员列表
eve::tensor::Graph类 参考

#include <Graph.h>

Public 成员函数

int addNode (GraphNode node)
 
const GraphNode & node (int id) const
 
GraphNode & node (int id)
 
int nodeCount () const
 
const std::vector< GraphNode > & nodes () const
 
std::vector< GraphNode > & nodes ()
 

静态 Public 成员函数

static int product (const int *dims, int rank)
 

详细描述

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

成员函数说明

◆ addNode()

int eve::tensor::Graph::addNode ( GraphNode  node)

在文件 Graph.cpp 第 25 行定义.

引用了 eve::tensor::GraphNode::dims, node(), product(), eve::tensor::GraphNode::rank , 以及 eve::tensor::GraphNode::size.

被这些函数引用 eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitBinary(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), eve::tensor::Func::emitConv1d(), eve::tensor::Func::emitConv2d(), eve::tensor::Func::emitEmbedding(), eve::tensor::Func::emitFill(), eve::tensor::Func::emitLayerNorm(), eve::tensor::Func::emitMatMul(), eve::tensor::Func::emitPermute(), eve::tensor::Func::emitPool(), eve::tensor::Func::emitReduce(), eve::tensor::Func::emitReshape(), eve::tensor::Func::emitResize2d(), eve::tensor::Func::emitRMSNorm(), eve::tensor::Func::emitSdpa(), eve::tensor::Func::emitSlice(), eve::tensor::Func::emitSoftmax(), eve::tensor::Func::emitTernary(), eve::tensor::Func::emitUnary(), eve::tensor::Func::emitUnaryScalar(), eve::tensor::Func::ensureNode(), eve::tensor::Func::input1(), eve::tensor::Func::input2(), eve::tensor::Func::input3(), eve::tensor::Func::input4(), eve::tensor::Func::input5() , 以及 eve::tensor::Func::input6().

◆ node() [1/2]

GraphNode & eve::tensor::Graph::node ( int  id)
inline

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

引用了 id.

◆ node() [2/2]

const GraphNode & eve::tensor::Graph::node ( int  id) const
inline

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

引用了 id.

被这些函数引用 addNode(), eve::tensor::Func::emitArgMax(), eve::tensor::Func::emitBinary(), eve::tensor::Func::emitCast(), eve::tensor::Func::emitConcat(), eve::tensor::Func::emitConv1d(), eve::tensor::Func::emitConv2d(), eve::tensor::Func::emitEmbedding(), eve::tensor::Func::emitLayerNorm(), eve::tensor::Func::emitMatMul(), eve::tensor::Func::emitPermute(), eve::tensor::Func::emitPool(), eve::tensor::Func::emitReduce(), eve::tensor::Func::emitReshape(), eve::tensor::Func::emitResize2d(), eve::tensor::Func::emitRMSNorm(), eve::tensor::Func::emitSdpa(), eve::tensor::Func::emitSlice(), eve::tensor::Func::emitSoftmax(), eve::tensor::Func::emitTernary(), eve::tensor::Func::emitTranspose(), eve::tensor::Func::emitUnary(), eve::tensor::Func::emitUnaryScalar() , 以及 eve::tensor::optimizeGraph().

◆ nodeCount()

int eve::tensor::Graph::nodeCount ( ) const
inline

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

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

◆ nodes() [1/2]

std::vector< GraphNode > & eve::tensor::Graph::nodes ( )
inline

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

◆ nodes() [2/2]

const std::vector< GraphNode > & eve::tensor::Graph::nodes ( ) const
inline

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

◆ product()

int eve::tensor::Graph::product ( const int *  dims,
int  rank 
)
static

在文件 Graph.cpp 第 16 行定义.

引用了 n.

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

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