|
enum | DataType { Generic,
Float,
Half,
Int8
} |
|
enum | Layout { Generic,
NCHW,
NHWC
} |
|
enum | NNDataKind { Generic,
Tensor
} |
| Discriminator for LLVM-style RTTI (isa<>)
|
|
enum | ValueKind { Value,
Instruction,
Data
} |
|
|
| Tensor (std::string name) |
|
NeuralNetData * | clone () |
|
void | setType (DataType type) |
|
DataType | getType () const |
|
const std::string | getName () const |
|
void | setName (const std::string &name) |
|
| NeuralNetData (NNDataKind kind) |
|
NNDataKind | getKind () const |
|
const std::string | getName () const |
|
size_t | getVersion () const |
|
void | setVersion (size_t version) |
|
| Value (ValueKind K) |
|
ValueKind | getKind () const |
|
Definition at line 158 of file NeuralNet.h.
The documentation for this class was generated from the following file:
- caffe2/core/nomnigraph/include/nomnigraph/Representations/NeuralNet.h