_eval(bool recompute) | magmadnn::op::DropoutOp< T > | protectedvirtual |
_grad(Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad) | magmadnn::op::DropoutOp< 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 |
copy (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | protected |
dropout_rate (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | protected |
DropoutOp(Operation< T > *input, float dropout_rate, unsigned long long seed, bool copy=true, bool needs_grad=true) (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | |
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 |
input (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | protected |
input_tensor (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | protected |
inputs (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | protected |
mask_tensor (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< 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 |
reset() | magmadnn::op::Operation< T > | inlinevirtual |
seed (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | protected |
to_string() | magmadnn::op::DropoutOp< T > | inlinevirtual |
~DropoutOp() (defined in magmadnn::op::DropoutOp< T >) | magmadnn::op::DropoutOp< T > | virtual |
~Operation() (defined in magmadnn::op::Operation< T >) | magmadnn::op::Operation< T > | inlinevirtual |