载入中...
搜索中...
未找到
Optimizer.h 文件参考
#include "common/Export.h"
#include "tensor/Graph.h"
#include <vector>

浏览源代码.

类

struct  eve::tensor::FusedGroup
 FusedGroup public API. 更多...
 
struct  eve::tensor::OptimizedGraph
 OptimizedGraph public API. 更多...
 

命名空间

namespace  eve
 Build metadata (engine git commit, build time, third-party version).
 
namespace  eve::tensor
 

枚举

enum class  eve::tensor::GroupKind : uint8_t {
  eve::tensor::Elementwise , eve::tensor::MatMul , eve::tensor::Conv1d , eve::tensor::Conv2d ,
  eve::tensor::MaxPool2d , eve::tensor::AvgPool2d , eve::tensor::Softmax , eve::tensor::LayerNorm ,
  eve::tensor::RMSNorm , eve::tensor::Reduce , eve::tensor::ArgMax , eve::tensor::Embedding ,
  eve::tensor::Concat , eve::tensor::Slice , eve::tensor::Permute , eve::tensor::Sdpa ,
  eve::tensor::Resize2d , eve::tensor::Alias
}
 GroupKind public API. 更多...
 

函数

OptimizedGraph eve::tensor::optimizeGraph (const Graph &graph, int outputNode)
 Optimize graph.
 
int eve::tensor::groupKernelCount (const OptimizedGraph &opt)
 Group kernel count.