nnvm::top::BatchNormParam Struct Reference

#include <nn.h>

Inheritance diagram for nnvm::top::BatchNormParam:

Collaboration diagram for nnvm::top::BatchNormParam:

Public Member Functions
	DMLC_DECLARE_PARAMETER (BatchNormParam)

Public Attributes
int	axis
double	epsilon
double	momentum
bool	center
bool	scale

Static Public Attributes
static const constexpr int	kData = 0
static const constexpr int	kGamma = 1
static const constexpr int	kBeta = 2
static const constexpr int	kMovingMean = 3
static const constexpr int	kMovingVariance = 4

Member Function Documentation

nnvm::top::BatchNormParam::DMLC_DECLARE_PARAMETER ( BatchNormParam  )

inline

Member Data Documentation

int nnvm::top::BatchNormParam::axis

bool nnvm::top::BatchNormParam::center

double nnvm::top::BatchNormParam::epsilon

const constexpr int nnvm::top::BatchNormParam::kBeta = 2

static

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

static

const constexpr int nnvm::top::BatchNormParam::kGamma = 1

static

const constexpr int nnvm::top::BatchNormParam::kMovingMean = 3

static

const constexpr int nnvm::top::BatchNormParam::kMovingVariance = 4

static

double nnvm::top::BatchNormParam::momentum

bool nnvm::top::BatchNormParam::scale

---

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

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