_eval(bool recompute=true) | magmadnn::op::ProductOp< T > | protectedvirtual |
_grad(Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad) | magmadnn::op::ProductOp< T > | protectedvirtual |
_grad_cache (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
a (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
a_tensor (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
add_consumer(Operation< T > *consumer) | magmadnn::op::Operation< T > | inlinevirtual |
alpha (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
b (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
b_tensor (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
consumers (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
copy (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
eval(bool recompute=true) | magmadnn::op::Operation< T > | inlinevirtual |
get_consumers() | magmadnn::op::Operation< T > | inlinevirtual |
get_grad_tensor(Operation< T > *wrt) | magmadnn::op::Operation< T > | inlinevirtual |
get_inputs() | magmadnn::op::Operation< T > | inlinevirtual |
get_memory_type() const | magmadnn::op::Operation< T > | inlinevirtual |
get_name() | magmadnn::op::Operation< T > | inlinevirtual |
get_output_shape() const | magmadnn::op::Operation< T > | inlinevirtual |
get_output_shape(unsigned int idx) const | magmadnn::op::Operation< T > | inlinevirtual |
get_output_size() const | magmadnn::op::Operation< T > | inlinevirtual |
get_output_tensor() | magmadnn::op::Operation< T > | inlinevirtual |
grad(Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad, bool recompute=true) | magmadnn::op::Operation< T > | inlinevirtual |
has_been_computed (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
has_grad_been_computed (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
inputs (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
mem_type (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
name (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
needs_grad (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
op_type (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | protected |
Operation() | magmadnn::op::Operation< T > | inline |
Operation(std::vector< Operation< T > *> inputs, bool needs_grad=true) (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | inline |
output_shape (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
output_tensor (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
ProductOp(T alpha, Operation< T > *a, Operation< T > *b, bool copy=true, bool needs_grad=true) (defined in magmadnn::op::ProductOp< T >) | magmadnn::op::ProductOp< T > | |
reset() | magmadnn::op::Operation< T > | inlinevirtual |
to_string() | magmadnn::op::ProductOp< T > | inlinevirtual |
~Operation() (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | inlinevirtual |