Go to the source code of this file.
|
struct | mshadow::expr::ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst > |
| reduce out the dimension of src labeled by axis. More...
|
|
struct | mshadow::expr::Plan< ReduceWithAxisExp< Reducer, SrcExp, DType, dimsrc, mask, dimdst >, DType > |
|
|
| 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
|
|
|
template<typename Reducer , bool mask, typename SrcExp , typename DType , int etype> |
ReduceWithAxisExp< Reducer, SrcExp, DType, ExpInfo< SrcExp >::kDim, mask, ExpInfo< SrcExp >::kDim - 1 > | mshadow::expr::reduce_with_axis (const Exp< SrcExp, DType, etype > &src, int axis) |
| reduce out the dimension of src labeled by axis. More...
|
|
template<typename Reducer , bool mask, typename SrcExp , typename DType , int etype> |
ReduceWithAxisExp< Reducer, SrcExp, DType, ExpInfo< SrcExp >::kDim, mask, ExpInfo< SrcExp >::kDim > | mshadow::expr::reduce_keepdim (const Exp< SrcExp, DType, etype > &src, int axis) |
| reduce out the dimension of src labeled by axis, keepdim turned on. More...
|
|