|
|
| Squeeze (Context &ctx, SqueezeConfig const &config=SqueezeConfig()) |
| |
|
virtual char const * | operator_type () const |
| | name of the operator type
|
| |
|
virtual Shape | new_shape (Shape const &in) |
| |
|
| ReshapeBase (Context const &) |
| |
|
virtual bool | alias_generator () |
| | returns true of the operator is alias - generation - it only changes the shape of tensor but not its content it actually does not perform any operation but only changes semantics, in this casse input and output are aliases of each other
|
| |
| virtual void | setup (std::vector< TensorSpecs > const &in, std::vector< TensorSpecs > &out, std::vector< TensorSpecs > ¶meters, size_t &workspace) |
| | Convigure operator. More...
|
| |
| virtual void | reshape (std::vector< Shape > const &in, std::vector< Shape > &out, size_t &ws) |
| | Reshape layer according to new input size. More...
|
| |
| virtual void | forward (std::vector< Tensor > &, std::vector< Tensor > &, std::vector< Tensor > &, Tensor &, ExecutionContext const &) |
| | Enqueue forward propogation computations. More...
|
| |
| virtual void | backward (std::vector< TensorAndGradient > &, std::vector< TensorAndGradient > &t, std::vector< TensorAndGradient > &, Tensor &, ExecutionContext const &) |
| | Enqueue backward propogation computations. More...
|
| |
|
| Operator (Context const &ctx) |
| | Create operator for specific context (device/platform)
|
| |
|
SharedResource & | shared_resource () |
| | Getter for object that is shared between operators accross the net, for example random numbers generator.
|
| |
|
void | shared_resource (std::shared_ptr< SharedResource > r) |
| | Setter of the shared resource.
|
| |
| virtual void | mode (CalculationsMode mode) |
| | Can be called with both train and predict before setup() is called. afterwards if original mode was train - it can be switched to predict and back but if original mode was predict it can't be switched to train. More...
|
| |
|
virtual CalculationsMode | mode () |
| | get current mode
|
| |
| | Operator (Operator const &)=delete |
| | operator non-copyable/moveable More...
|
| |
|
void | operator= (Operator const &)=delete |
| |
|
| Operator (Operator &&)=delete |
| |
|
void | operator= (Operator &&)=delete |
| |
|
virtual void | initialize_params (std::vector< Tensor > &, ExecutionContext const &) |
| | Set default parameters iniitalization.
|
| |
Public Member Functions inherited from dlprim::ReshapeBase