mxnet
|
#include <nn.h>
Public Member Functions | |
DMLC_DECLARE_PARAMETER (AvgPool2DParam) | |
Public Attributes | |
TShape | pool_size |
TShape | strides |
TShape | padding |
std::string | layout |
bool | ceil_mode |
bool | count_include_pad |
|
inline |
bool nnvm::top::AvgPool2DParam::ceil_mode |
bool nnvm::top::AvgPool2DParam::count_include_pad |
std::string nnvm::top::AvgPool2DParam::layout |
TShape nnvm::top::AvgPool2DParam::padding |
TShape nnvm::top::AvgPool2DParam::pool_size |
TShape nnvm::top::AvgPool2DParam::strides |