#include "../extension.h"

Include dependency graph for take_grad.h:

This graph shows which files directly or indirectly include this file:

Classes
struct	mshadow::expr::TakeGradExp< IndexExp, SrcExp, DType >
	Calculate embedding gradient. More...

struct	mshadow::expr::Plan< TakeGradExp< IndexExp, SrcExp, DType >, DType >

struct	mshadow::expr::ShapeCheck< dim, TakeGradExp< IndexExp, SrcExp, DType > >

struct	mshadow::expr::ExpInfo< TakeGradExp< IndexExp, SrcExp, DType > >

Namespaces
	mshadow
	overloaded + operator between half_t and bf16_t

	mshadow::expr
	namespace for abstract expressions and expressions template, have no dependency on tensor.h, These data structure takes no charge in computations, they are only used to define operations and represent expression in a symbolic way

Functions
template<typename IndexExp , typename SrcExp , typename DType , int e1, int e2>
TakeGradExp< IndexExp, SrcExp, DType >	mshadow::expr::take_grad (const Exp< IndexExp, DType, e1 > &index, const Exp< SrcExp, DType, e2 > &src, const index_t input_dim)

template<typename IndexExp , typename SrcExp , typename DType >
Plan< TakeGradExp< IndexExp, SrcExp, DType >, DType >	mshadow::expr::MakePlan (const TakeGradExp< IndexExp, SrcExp, DType > &exp)

Detailed Description

Classes