Searched refs:kMomentumOriginIdx (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 216 const OptimOriginIdx kMomentumOriginIdx = {{"weight", 0}, {"accum", 1}, {"lr", 2}, {"grad", 3}, {"m… variable
239 …st std::map<std::string, OptimOriginIdx> kOptimToOriginIdx = {{kApplyMomentum, kMomentumOriginIdx},
|
| D | optimizer_info.cc | 287 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in gradient()
294 size_t origin_grad_index = kMomentumOriginIdx.at("grad"); in indices()
|