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

Go to the source code of this file.

Classes

struct  mshadow::expr::ChannelUnpoolingExp< Reducer, SrcExp, DType, srcdim >
 channel pooling expression, do reduction over (local nearby) channels, used to implement local response normalization More...
 
struct  mshadow::expr::Plan< ChannelUnpoolingExp< Reducer, SrcExp, DType, srcdim >, DType >
 

Namespaces

 mshadow
 namespace for mshadow
 
 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 , typename SrcExp , typename DType , int etype>
ChannelUnpoolingExp< Reducer, SrcExp, DType, ExpInfo< SrcExp >::kDim > mshadow::expr::ch_unpool (const Exp< SrcExp, DType, etype > &data_src, const Exp< SrcExp, DType, etype > &data_pooled, const Exp< SrcExp, DType, etype > &grad_pooled, index_t nsize, index_t stride, index_t pad)
 channel unpooling, do unroll over (local nearby) channels More...
 
template<typename Reducer , typename SrcExp , typename DType , int etype>
ChannelUnpoolingExp< Reducer, SrcExp, DType, ExpInfo< SrcExp >::kDim > mshadow::expr::ch_unpool (const Exp< SrcExp, DType, etype > &data_src, const Exp< SrcExp, DType, etype > &data_pooled, const Exp< SrcExp, DType, etype > &grad_pooled, index_t nsize)