Public Member Functions | |
GenericOperator (std::string name) | |
NOMNIGRAPH_DEFINE_NN_RTTI (GenericOperator) | |
std::string | getName () const |
void | setName (std::string name) |
Public Member Functions inherited from nom::repr::NeuralNetOperator | |
NeuralNetOperator (NNKind K, Opcode I, NNLayout L) | |
NeuralNetOperator (NNKind K, Opcode I) | |
NeuralNetOperator (NNKind K, NNLayout L) | |
NeuralNetOperator (NNKind K) | |
NNKind | getKind () const |
void | setLayout (NNLayout L) |
NNLayout | getLayout () const |
void | setAnnotation (std::unique_ptr< Annotation > extraAnnotation) |
const Annotation * | getAnnotation () const |
Annotation * | getMutableAnnotation () |
const std::string | getName () const |
bool | checkInputsAndOutputs (std::vector< const NeuralNetData * > inputs, std::vector< const NeuralNetData * > outputs) |
Validate the inputs and outputs to this operator. More... | |
NeuralNetOperator (const NeuralNetOperator &)=delete | |
NeuralNetOperator & | operator= (NeuralNetOperator &)=delete |
Public Member Functions inherited from nom::repr::Instruction | |
Instruction (Opcode op) | |
Opcode | getOpcode () const |
Public Member Functions inherited from nom::repr::Value | |
Value (ValueKind K) | |
ValueKind | getKind () const |
Additional Inherited Members | |
Public Types inherited from nom::repr::NeuralNetOperator | |
enum | NNKind { Undefined, GenericOperator, NNPhi, While } |
Discriminator for LLVM-style RTTI (isa<>) | |
enum | NNLayout { Undefined, NCHW, NHWC } |
An optional tensor-type specifier. | |
Public Types inherited from nom::repr::Instruction | |
enum | Opcode { Generic, TerminatorStart, Branch, Return, TerminatorEnd, Phi } |
All the different types of execution. | |
Public Types inherited from nom::repr::Value | |
enum | ValueKind { Value, Instruction, Data } |
Static Public Member Functions inherited from nom::repr::Instruction | |
static bool | classof (const Value *V) |
Definition at line 225 of file NeuralNet.h.