Searched refs:kMomentum (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | bn_grad.cc | 37 void BNGrad::set_momentum(const float momentum) { (void)this->AddAttr(kMomentum, MakeValue(momentum… in set_momentum()
40 auto value_ptr = this->GetAttr(kMomentum); in get_momentum()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | fused_batch_norm.cc | 34 void FusedBatchNorm::set_momentum(const float momentum) { (void)this->AddAttr(kMomentum, MakeValue(… in set_momentum()
47 auto value_ptr = this->GetAttr(kMomentum); in get_momentum()
|
| D | batch_norm.cc | 47 …CheckAndConvertUtils::CheckInRange<float>(kMomentum, momentun, kIncludeBoth, {0.0, 1.0}, this->nam… in set_momentum()
48 (void)this->AddAttr(kMomentum, MakeValue(momentun)); in set_momentum()
52 auto value_ptr = GetAttr(kMomentum); in get_momentum()
|
| D | op_utils.h | 119 constexpr auto kMomentum = "momentum"; variable
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | common.h | 121 constexpr auto kMomentum = "momentum"; variable
139 …{"inputs", {{kWeight, 0}, {kAccumulation, 1}, {kLearningRate, 2}, {kGradient, 3}, {kMomentum, 4}}}…
179 …{"inputs", {{kWeight, 0}, {kGradient, 1}, {kLearningRate, 2}, {kAccumulation, 3}, {kMomentum, 4}, …
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/kernel/ |
| D | apply_momentum_kernel.cc | 29 .AddInputNameType(kMomentum, kNumberTypeFloat32),
|
| D | sgd_kernel.cc | 29 .AddInputNameType(kMomentum, kNumberTypeFloat32)
|
| D | apply_momentum_kernel.h | 55 reuse_kernel_node_inputs_info_.insert(std::make_pair(kMomentum, 4)); in GenerateReuseKernelNodeInfo()
|
| D | sgd_kernel.h | 54 reuse_kernel_node_inputs_info_.insert(std::make_pair(kMomentum, 4)); in GenerateReuseKernelNodeInfo()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 48 constexpr char kMomentum[] = "momentum"; variable
|