_eval(bool recompute=true) | magmadnn::op::Variable< T > | protectedvirtual |
_grad(Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad) | magmadnn::op::Variable< T > | protectedvirtual |
_grad_cache (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
add_consumer(Operation< T > *consumer) | magmadnn::op::Operation< T > | inlinevirtual |
consumers (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
delete_tensor (defined in magmadnn::op::Variable< T >) | magmadnn::op::Variable< 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::Variable< 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::Variable< T >) | magmadnn::op::Variable< T > | protected |
needs_grad (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< 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 |
reset() | magmadnn::op::Operation< T > | inlinevirtual |
to_string() | magmadnn::op::Variable< T > | inlinevirtual |
val (defined in magmadnn::op::Variable< T >) | magmadnn::op::Variable< T > | protected |
Variable(std::string name, std::vector< unsigned int > shape, tensor_filler_t< T > filler, memory_t mem_type) (defined in magmadnn::op::Variable< T >) | magmadnn::op::Variable< T > | |
Variable(std::string name, Tensor< T > *val) (defined in magmadnn::op::Variable< T >) | magmadnn::op::Variable< T > | |
~Operation() (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | inlinevirtual |
~Variable() (defined in magmadnn::op::Variable< T >) | magmadnn::op::Variable< T > | |