mxnet
|
#include <nn.h>
Public Member Functions | |
DMLC_DECLARE_PARAMETER (LayoutTransformParam) | |
Public Attributes | |
std::string | src_layout |
std::string | dst_layout |
|
inline |
std::string nnvm::top::LayoutTransformParam::dst_layout |
std::string nnvm::top::LayoutTransformParam::src_layout |