nnvm::top::Conv2DParam Struct Reference

#include <nn.h>

Inheritance diagram for nnvm::top::Conv2DParam:

Collaboration diagram for nnvm::top::Conv2DParam:

Public Member Functions
	DMLC_DECLARE_PARAMETER (Conv2DParam)

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

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::Conv2DParam::DMLC_DECLARE_PARAMETER ( Conv2DParam  )

inline

Member Data Documentation

int nnvm::top::Conv2DParam::channels

TShape nnvm::top::Conv2DParam::dilation

int nnvm::top::Conv2DParam::groups

const constexpr int nnvm::top::Conv2DParam::kBias = 2

static

const constexpr int nnvm::top::Conv2DParam::kData = 0

static

std::string nnvm::top::Conv2DParam::kernel_layout

TShape nnvm::top::Conv2DParam::kernel_size

const constexpr int nnvm::top::Conv2DParam::kWeight = 1

static

std::string nnvm::top::Conv2DParam::layout

int nnvm::top::Conv2DParam::out_dtype

std::string nnvm::top::Conv2DParam::out_layout

TShape nnvm::top::Conv2DParam::padding

TShape nnvm::top::Conv2DParam::strides

bool nnvm::top::Conv2DParam::use_bias

---

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

• /work/mxnet/3rdparty/tvm/nnvm/include/nnvm/top/nn.h