|
virtual void | setGatherAxis (int32_t axis)=0 |
| Set the axis to gather on. The axis must be less than the number of dimensions in the data input. More...
|
|
virtual int32_t | getGatherAxis () const =0 |
| Get the axis to gather on. More...
|
|
virtual void | setNbElementWiseDims (int32_t k)=0 |
| Set the number of leading dimensions of indices tensor to be handled elementwise. k must be 0 if there is an implicit batch dimension. It can be 0 or 1 if there is not an implicit batch dimension. More...
|
|
virtual int32_t | getNbElementWiseDims () const =0 |
| Get the number of leading dimensions of indices tensor to be handled elementwise. More...
|
|
virtual LayerType | getType () const =0 |
| Return the type of a layer. More...
|
|
virtual void | setName (const char *name)=0 |
| Set the name of a layer. More...
|
|
virtual const char * | getName () const =0 |
| Return the name of a layer. More...
|
|
virtual int32_t | getNbInputs () const =0 |
| Get the number of inputs of a layer.
|
|
virtual ITensor * | getInput (int32_t index) const =0 |
| Get the layer input corresponding to the given index. More...
|
|
virtual int32_t | getNbOutputs () const =0 |
| Get the number of outputs of a layer.
|
|
virtual ITensor * | getOutput (int32_t index) const =0 |
| Get the layer output corresponding to the given index. More...
|
|
virtual void | setInput (int32_t index, ITensor &tensor)=0 |
| Replace an input of this layer with a specific tensor. More...
|
|
virtual void | setPrecision (DataType dataType)=0 |
| Set the computational precision of this layer. More...
|
|
virtual DataType | getPrecision () const =0 |
| get the computational precision of this layer More...
|
|
virtual bool | precisionIsSet () const =0 |
| whether the computational precision has been set for this layer More...
|
|
virtual void | resetPrecision ()=0 |
| reset the computational precision for this layer More...
|
|
virtual void | setOutputType (int32_t index, DataType dataType)=0 |
| Set the output type of this layer. More...
|
|
virtual DataType | getOutputType (int32_t index) const =0 |
| get the output type of this layer More...
|
|
virtual bool | outputTypeIsSet (int32_t index) const =0 |
| whether the output type has been set for this layer More...
|
|
virtual void | resetOutputType (int32_t index)=0 |
| reset the output type for this layer More...
|
|