Searched defs:BatchNormParam (Results 1 – 4 of 4) sorted by relevance
68 struct BatchNormParam : public dmlc::Parameter<BatchNormParam> { struct80 DMLC_DECLARE_PARAMETER(BatchNormParam) { in DMLC_DECLARE_PARAMETER() argument134 struct hash<mxnet::op::BatchNormParam> { argument
63 struct BatchNormParam : public dmlc::Parameter<BatchNormParam> { struct70 DMLC_DECLARE_PARAMETER(BatchNormParam) { in DMLC_DECLARE_PARAMETER() argument85 static const constexpr int kData = 0;86 static const constexpr int kGamma = 1;87 static const constexpr int kBeta = 2;88 static const constexpr int kMovingMean = 3;89 static const constexpr int kMovingVariance = 4;