magmadnn::op::Operation< T > Class Template Reference

Inheritance diagram for magmadnn::op::Operation< T >:

Collaboration diagram for magmadnn::op::Operation< T >:

Public Member Functions
	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	to_string ()=0
virtual std::string	get_name ()

Protected Member Functions
virtual Tensor< T > *	_eval (bool recompute=true)=0
virtual Tensor< T > *	_grad (Operation< T > *consumer, Operation< T > *var, Tensor< T > *grad)=0

Protected Attributes
std::vector< Operation< T > * >	inputs
std::vector< Operation< T > * >	consumers
std::vector< unsigned int >	output_shape
memory_t	mem_type
std::map< uintptr_t, Tensor< T > * >	_grad_cache
std::string	name = "DefaultOpName"
Tensor< T > *	output_tensor
bool	needs_grad
bool	has_been_computed
bool	has_grad_been_computed

Constructor & Destructor Documentation

Operation()

template<typename T>

magmadnn::op::Operation< T >::Operation ( )

inline

The operation class serves as an abstract object, which all tensors operations descend from. It is used to build a computation tree.

Member Function Documentation

_eval()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::_eval ( bool  recompute = true )

protectedpure virtual

Sets this->output_tensor to the value of this operation

Returns

Tensor<T>* the evaluated tensor

Implemented in magmadnn::op::ProductOp< T >, magmadnn::op::DivOp< T >, magmadnn::op::AddOp< T >, magmadnn::op::SigmoidOp< T >, magmadnn::op::MatmulOp< T >, magmadnn::op::Variable< T >, magmadnn::op::SumOp< T >, magmadnn::op::TanhOp< T >, magmadnn::op::ReluOp< T >, magmadnn::op::CrossEntropyOp< T >, magmadnn::op::DropoutOp< T >, magmadnn::op::SoftmaxOp< T >, magmadnn::op::ReduceSumOp< T >, magmadnn::op::ScalarProductOp< T >, magmadnn::op::LinearForwardOp< T >, magmadnn::op::Conv2DForwardOp< T >, magmadnn::op::LogOp< T >, magmadnn::op::PowOp< T >, magmadnn::op::NegativeOp< T >, magmadnn::op::PoolingOp< T >, magmadnn::op::TransposeOp< T >, and magmadnn::op::FlattenOp< T >.

Here is the caller graph for this function:

_grad()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::_grad ( Operation< T > *  consumer, Operation< T > *  var, Tensor< T > *  grad  )

protectedpure virtual

Computes the gradient of this operation wrt the output of consumer.

Parameters

consumer
var
grad
recompute

Returns

Tensor<T>*

Implemented in magmadnn::op::ProductOp< T >, magmadnn::op::DivOp< T >, magmadnn::op::AddOp< T >, magmadnn::op::SigmoidOp< T >, magmadnn::op::MatmulOp< T >, magmadnn::op::Variable< T >, magmadnn::op::SumOp< T >, magmadnn::op::TanhOp< T >, magmadnn::op::ReluOp< T >, magmadnn::op::CrossEntropyOp< T >, magmadnn::op::DropoutOp< T >, magmadnn::op::SoftmaxOp< T >, magmadnn::op::ReduceSumOp< T >, magmadnn::op::ScalarProductOp< T >, magmadnn::op::LinearForwardOp< T >, magmadnn::op::Conv2DForwardOp< T >, magmadnn::op::LogOp< T >, magmadnn::op::PowOp< T >, magmadnn::op::NegativeOp< T >, magmadnn::op::PoolingOp< T >, magmadnn::op::TransposeOp< T >, and magmadnn::op::FlattenOp< T >.

Here is the caller graph for this function:

add_consumer()

template<typename T>

virtual void magmadnn::op::Operation< T >::add_consumer ( Operation< T > *  consumer )

inlinevirtual

Parameters

consumer

eval()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::eval ( bool  recompute = true )

inlinevirtual

Returns the operation's evaluated tensor.

Parameters

recompute

whether to use previous value or recalculate

Returns

Tensor<T>*

Here is the call graph for this function:

Here is the caller graph for this function:

get_consumers()

template<typename T>

virtual std::vector<Operation<T> *> magmadnn::op::Operation< T >::get_consumers ( )

inlinevirtual

Returns a vector of operations that need this operation as input.

Returns

std::vector<Operation<T> *> vector of consumer operations

get_grad_tensor()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::get_grad_tensor ( Operation< T > *  wrt )

inlinevirtual

Gets the current grad_tensor wrt to wrt.

Parameters

wrt

Returns

Tensor<T>*

Here is the call graph for this function:

get_inputs()

template<typename T>

virtual std::vector<Operation<T> *> magmadnn::op::Operation< T >::get_inputs ( )

inlinevirtual

Returns a vector the input operations to this one.

Returns

std::vector<Operation<T> *> vector of input operations

get_memory_type()

template<typename T>

virtual memory_t magmadnn::op::Operation< T >::get_memory_type ( ) const

inlinevirtual

The memory type used to compute this operation.

Returns

memory_t

get_name()

template<typename T>

virtual std::string magmadnn::op::Operation< T >::get_name ( )

inlinevirtual

the name of this operation.

Returns

std::string

Reimplemented in magmadnn::op::Variable< T >.

Here is the call graph for this function:

get_output_shape() [1/2]

template<typename T>

virtual std::vector<unsigned int> magmadnn::op::Operation< T >::get_output_shape ( ) const

inlinevirtual

Returns the expected output shape of this operation.

Returns

std::vector<unsigned int>

get_output_shape() [2/2]

template<typename T>

virtual unsigned int magmadnn::op::Operation< T >::get_output_shape ( unsigned int  idx ) const

inlinevirtual

Parameters

idx

Returns

std::vector<unsigned int>

get_output_size()

template<typename T>

virtual unsigned int magmadnn::op::Operation< T >::get_output_size ( ) const

inlinevirtual

The total number of elements outputted by operation.

Returns

unsigned int

get_output_tensor()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::get_output_tensor ( )

inlinevirtual

Gets a pointer to the output tensor this returns

Returns

Tensor<T>*

grad()

template<typename T>

virtual Tensor<T>* magmadnn::op::Operation< T >::grad ( Operation< T > *  consumer, Operation< T > *  var, Tensor< T > *  grad, bool  recompute = true  )

inlinevirtual

Computes the gradient with respect to the outputs and var.

Parameters

consumer	the operation that consumes this that needs the gradient
grad	the gradient of the loss w.r.t. the consumers output

Returns

Tensor<T>*

Here is the call graph for this function:

Here is the caller graph for this function:

reset()

template<typename T>

virtual void magmadnn::op::Operation< T >::reset ( )

inlinevirtual

Clears the operation so that it will be recomputed.

to_string()

template<typename T>

virtual std::string magmadnn::op::Operation< T >::to_string ( )

pure virtual

string form of the given operation. Expands on input.

Returns

std::string

Implemented in magmadnn::op::ProductOp< T >, magmadnn::op::DivOp< T >, magmadnn::op::AddOp< T >, magmadnn::op::SigmoidOp< T >, magmadnn::op::MatmulOp< T >, magmadnn::op::Variable< T >, magmadnn::op::SumOp< T >, magmadnn::op::TanhOp< T >, magmadnn::op::ReluOp< T >, magmadnn::op::CrossEntropyOp< T >, magmadnn::op::DropoutOp< T >, magmadnn::op::SoftmaxOp< T >, magmadnn::op::ReduceSumOp< T >, magmadnn::op::ScalarProductOp< T >, magmadnn::op::LinearForwardOp< T >, magmadnn::op::Conv2DForwardOp< T >, magmadnn::op::LogOp< T >, magmadnn::op::PowOp< T >, magmadnn::op::NegativeOp< T >, magmadnn::op::PoolingOp< T >, magmadnn::op::TransposeOp< T >, and magmadnn::op::FlattenOp< T >.

Here is the caller graph for this function:

---

The documentation for this class was generated from the following file:

• include/compute/operation.h