Input objects for the forward batch normalization layer.
More...
◆ Input() [1/2]
◆ Input() [2/2]
◆ allocate()
Allocates memory to store input objects of forward batch normalization layer
- Parameters
-
[in] | parameter | Parameter of forward batch normalization layer |
[in] | method | Computation method for the layer |
- Returns
- Status of computations
◆ check()
Checks input object of the forward batch normalization layer
- Parameters
-
[in] | parameter | Parameter of layer |
[in] | method | Computation method of the layer |
- Returns
- Status of computations
Reimplemented from Input.
◆ get()
Returns an input object for forward batch normalization layer
- Parameters
-
[in] | id | Identifier of the input object |
- Returns
- Input object that corresponds to the given identifier
◆ getBiasesSizes()
Returns dimensions of biases tensor
- Returns
- Dimensions of biases tensor
Reimplemented from Input.
◆ getWeightsSizes()
Returns dimensions of weights tensor
- Returns
- Dimensions of weights tensor
Reimplemented from Input.
◆ set()
Sets input for the forward batch normalization layer
- Parameters
-
[in] | id | Identifier of the input object |
[in] | ptr | Input object to set |
The documentation for this class was generated from the following file: