mxnet
Classes | Namespaces | Functions
reduce_with_axis.h File Reference
#include "../extension.h"
Include dependency graph for reduce_with_axis.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

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 >
 

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 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...
 

Detailed Description

Copyright (c) 2015 by Contributors

Author
Junyuan Xie