Searched refs:kLearningRate (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | common.h | 118 constexpr auto kLearningRate = "lr"; variable
139 …{"inputs", {{kWeight, 0}, {kAccumulation, 1}, {kLearningRate, 2}, {kGradient, 3}, {kMomentum, 4}}}…
146 {kLearningRate, 5},
158 {kLearningRate, 5},
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 | 27 .AddInputNameType(kLearningRate, kNumberTypeFloat32)
|
| D | sgd_kernel.cc | 27 .AddInputNameType(kLearningRate, kNumberTypeFloat32)
|
| D | apply_momentum_kernel.h | 54 reuse_kernel_node_inputs_info_.insert(std::make_pair(kLearningRate, 2)); in GenerateReuseKernelNodeInfo()
|
| D | sgd_kernel.h | 52 reuse_kernel_node_inputs_info_.insert(std::make_pair(kLearningRate, 2)); in GenerateReuseKernelNodeInfo()
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 47 constexpr char kLearningRate[] = "learning_rate"; variable
|