Graph.h 文件参考

#include "tensor/Tensor.h"
#include <cstdint>
#include <memory>
#include <string>
#include <vector>

浏览源代码.

类
struct	eve::tensor::GraphNode
class	eve::tensor::Graph
class	eve::tensor::Func
	Trace builder — TF2 tf.function analogue (tf.func in scripts). While active, TF ops record into this graph. 更多...
class	eve::tensor::CompiledFunction
	Optimized / scheduled graph ready to run with feeds. 更多...

命名空间
namespace	eve
namespace	eve::tensor

枚举
enum class	eve::tensor::OpType : uint8_t {   eve::tensor::Placeholder = 0 , eve::tensor::Const , eve::tensor::Add , eve::tensor::Sub ,   eve::tensor::Multiply , eve::tensor::Divide , eve::tensor::AddScalar , eve::tensor::SubScalar ,   eve::tensor::MulScalar , eve::tensor::DivScalar , eve::tensor::Neg , eve::tensor::Abs ,   eve::tensor::Sqrt , eve::tensor::Exp , eve::tensor::Log , eve::tensor::Sin ,   eve::tensor::Cos , eve::tensor::Tanh , eve::tensor::Relu , eve::tensor::Sigmoid ,   eve::tensor::Gelu , eve::tensor::Silu , eve::tensor::PowScalar , eve::tensor::Clamp ,   eve::tensor::MaximumScalar , eve::tensor::MinimumScalar , eve::tensor::Where , eve::tensor::MatMul ,   eve::tensor::Transpose , eve::tensor::Permute , eve::tensor::Reshape , eve::tensor::Flatten ,   eve::tensor::Softmax , eve::tensor::LogSoftmax , eve::tensor::LayerNorm , eve::tensor::RMSNorm ,   eve::tensor::Conv1d , eve::tensor::Conv2d , eve::tensor::MaxPool2d , eve::tensor::AvgPool2d ,   eve::tensor::Embedding , eve::tensor::Concat , eve::tensor::Slice , eve::tensor::ReduceSum ,   eve::tensor::ReduceMean , eve::tensor::ReduceMin , eve::tensor::ReduceMax , eve::tensor::ArgMax ,   eve::tensor::Cast , eve::tensor::ScaledDotProductAttention , eve::tensor::Resize2d }