nom::repr::Conv Class Reference

Inheritance diagram for nom::repr::Conv:

Public Member Functions
	Conv (std::vector< int > kernelShape, std::vector< int > dilations={1, 1}, int group=1, std::vector< int > pads={0, 0}, std::vector< int > strides={1, 1})
	NOMNIGRAPH_DEFINE_NN_RTTI (Conv)
void	setDilations (std::vector< int > &&dilations)
void	setGroup (int &&group)
void	setPads (std::vector< int > &&pads)
void	setStrides (std::vector< int > &&strides)
std::vector< int >	getDilations ()
int	getGroup ()
std::vector< int >	getPads ()
std::vector< int >	getStrides ()
std::vector< int >	getKernelShape ()
bool	checkInputsAndOutputs (std::vector< const NeuralNetData * > inputs, std::vector< const NeuralNetData * > outputs)
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

Protected Attributes
std::vector< int >	KernelShape
std::vector< int >	Dilations
int	Group
std::vector< int >	Pads
std::vector< int >	Strides

Additional Inherited Members
Public Types inherited from nom::repr::NeuralNetOperator
enum	NNKind {   Undefined, Conv, Relu, ConvRelu,   DynamicInput, Send, Receive, While,   NNPhi, GenericOperator }
	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)

Detailed Description

Definition at line 189 of file NeuralNet.h.

---

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

• caffe2/core/nomnigraph/include/nomnigraph/Representations/NeuralNet.h