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: