Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Types | Public Member Functions | Static Public Member Functions
nom::repr::Tensor Class Reference
Inheritance diagram for nom::repr::Tensor:
nom::repr::NeuralNetData nom::repr::Data nom::repr::Value

Public Types

enum  DataType { Generic, Float, Half, Int8 }
enum  Layout { Generic, NCHW, NHWC }
- Public Types inherited from nom::repr::NeuralNetData
enum  NNDataKind { Generic, Tensor }
 Discriminator for LLVM-style RTTI (isa<>)
- Public Types inherited from nom::repr::Value
enum  ValueKind { Value, Instruction, Data }

Public Member Functions

 Tensor (std::string name)
NeuralNetDataclone ()
void setType (DataType type)
DataType getType () const
const std::string getName () const
void setName (const std::string &name)
- Public Member Functions inherited from nom::repr::NeuralNetData
 NeuralNetData (NNDataKind kind)
NNDataKind getKind () const
const std::string getName () const
- Public Member Functions inherited from nom::repr::Data
size_t getVersion () const
void setVersion (size_t version)
- Public Member Functions inherited from nom::repr::Value
 Value (ValueKind K)
ValueKind getKind () const

Static Public Member Functions

static bool classof (const NeuralNetData *D)
- Static Public Member Functions inherited from nom::repr::Data
static bool classof (const Value *V)

Detailed Description

Definition at line 158 of file NeuralNet.h.

The documentation for this class was generated from the following file: