mxnet
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
nnvm::top::WinogradConv2DParam Struct Reference

#include <nn.h>

Inheritance diagram for nnvm::top::WinogradConv2DParam:
Inheritance graph
Collaboration diagram for nnvm::top::WinogradConv2DParam:
Collaboration graph

Public Member Functions

 DMLC_DECLARE_PARAMETER (WinogradConv2DParam)
 

Public Attributes

int channels
 
TShape kernel_size
 
TShape strides
 
TShape padding
 
TShape dilation
 
int groups
 
std::string layout
 
std::string kernel_layout
 
std::string out_layout
 
int out_dtype
 
bool use_bias
 
int tile_size
 

Static Public Attributes

static const constexpr int kData = 0
 
static const constexpr int kWeight = 1
 
static const constexpr int kBias = 2
 

Member Function Documentation

nnvm::top::WinogradConv2DParam::DMLC_DECLARE_PARAMETER ( WinogradConv2DParam  )
inline

Member Data Documentation

int nnvm::top::WinogradConv2DParam::channels
TShape nnvm::top::WinogradConv2DParam::dilation
int nnvm::top::WinogradConv2DParam::groups
const constexpr int nnvm::top::WinogradConv2DParam::kBias = 2
static
const constexpr int nnvm::top::WinogradConv2DParam::kData = 0
static
std::string nnvm::top::WinogradConv2DParam::kernel_layout
TShape nnvm::top::WinogradConv2DParam::kernel_size
const constexpr int nnvm::top::WinogradConv2DParam::kWeight = 1
static
std::string nnvm::top::WinogradConv2DParam::layout
int nnvm::top::WinogradConv2DParam::out_dtype
std::string nnvm::top::WinogradConv2DParam::out_layout
TShape nnvm::top::WinogradConv2DParam::padding
TShape nnvm::top::WinogradConv2DParam::strides
int nnvm::top::WinogradConv2DParam::tile_size
bool nnvm::top::WinogradConv2DParam::use_bias

The documentation for this struct was generated from the following file: