|
| ScalarProductOp (T alpha, Operation< T > *x, bool copy=true, bool needs_grad=true) |
|
| ScalarProductOp (Operation< T > *scalar, Operation< T > *x, bool copy=true, bool needs_grad=true) |
|
std::string | to_string () |
|
| Operation () |
|
| Operation (std::vector< Operation< T > *> inputs, bool needs_grad=true) |
|
virtual std::vector< unsigned int > | get_output_shape () const |
|
virtual unsigned int | get_output_shape (unsigned int idx) const |
|
virtual unsigned int | get_output_size () const |
|
virtual memory_t | get_memory_type () const |
|
virtual Tensor< T > * | eval (bool recompute=true) |
|
virtual void | reset () |
|
virtual Tensor< T > * | grad (Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad, bool recompute=true) |
|
virtual void | add_consumer (Operation< T > *consumer) |
|
virtual std::vector< Operation< T > * > | get_consumers () |
|
virtual std::vector< Operation< T > * > | get_inputs () |
|
virtual Tensor< T > * | get_output_tensor () |
|
virtual Tensor< T > * | get_grad_tensor (Operation< T > *wrt) |
|
virtual std::string | get_name () |
|
template<typename T>
class magmadnn::op::ScalarProductOp< T >
Multiplies a tensor by a scalar
- Template Parameters
-