_eval(bool recompute) | magmadnn::op::Conv2DForwardOp< T > | protectedvirtual |
_grad(Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad) | magmadnn::op::Conv2DForwardOp< 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 |
calculate_and_set_output_shape() (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
consumers (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
Conv2DForwardOp(Operation< T > *input, Operation< T > *filter, int pad_h=0, int pad_w=0, int vertical_stride=1, int horizontal_stride=1, int dilation_h=1, int dilation_w=1, bool use_cross_correlation=true, bool needs_grad=true) (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | |
dilation_h (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
dilation_w (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
eval(bool recompute=true) | magmadnn::op::Operation< T > | inlinevirtual |
filter (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
filter_tensor (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
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 |
horizontal_stride (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
init_settings() (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
input (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
input_tensor (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< 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 |
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 |
pad_h (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
pad_w (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
reset() | magmadnn::op::Operation< T > | inlinevirtual |
to_string() | magmadnn::op::Conv2DForwardOp< T > | inlinevirtual |
use_cross_correlation (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
vertical_stride (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | protected |
~Conv2DForwardOp() (defined in magmadnn::op::Conv2DForwardOp< T >) | magmadnn::op::Conv2DForwardOp< T > | |
~Operation() (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | inlinevirtual |