mxnet
extension Directory Reference
Directory dependency graph for extension:
/work/mxnet/3rdparty/mshadow/mshadow/extension

Files

file  broadcast.h [code]
 support for broadcast and repmat
 
file  broadcast_with_axis.h [code]
 
file  channel_pool.h [code]
 support for chpool
 
file  channel_unpool.h [code]
 
file  choose.h [code]
 support for implicit array selection operation
 
file  complex.h [code]
 support for complex operations
 
file  concat.h [code]
 support for concatenation
 
file  crop.h [code]
 support for crop
 
file  fill.h [code]
 support for implicit array filling operation
 
file  flip.h [code]
 support for flip a certain dimension.
 
file  implicit_gemm.h [code]
 support for implicit GEMM operation
 
file  mask.h [code]
 
file  mirror.h [code]
 support for mirror
 
file  one_hot.h [code]
 Create one-hot indicator array based on the index.
 
file  pack_col2patch.h [code]
 support for pack
 
file  pad.h [code]
 support for pad
 
file  range.h [code]
 support generating a range vector
 
file  reduce_with_axis.h [code]
 
file  reduceto1d.h [code]
 support for sum_rows and sumall_except_dim
 
file  reshape.h [code]
 support for reshape
 
file  slice.h [code]
 support for slice a certain dimension.
 
file  slice_ex.h [code]
 
file  spatial_pool.h [code]
 support for spatial pooling
 
file  spatial_unpool.h [code]
 support for unpool
 
file  spatial_upsampling_nearest.h [code]
 
file  swapaxis.h [code]
 support for swapaxis
 
file  take.h [code]
 
file  take_grad.h [code]
 
file  transpose.h [code]
 support for transpose
 
file  unpack_patch2col.h [code]
 support for unpack