mxnet
|
#include <tensor.h>
Public Member Functions | |
DMLC_DECLARE_PARAMETER (SplitParam) | |
Public Attributes | |
Tuple< int > | indices_or_sections |
int | axis |
bool | equal_split |
|
inline |
int nnvm::top::SplitParam::axis |
bool nnvm::top::SplitParam::equal_split |
Tuple<int> nnvm::top::SplitParam::indices_or_sections |