Searched refs:OptimOriginIdx (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 211 using OptimOriginIdx = std::map<std::string, size_t>; variable
216 const OptimOriginIdx kMomentumOriginIdx = {{"weight", 0}, {"accum", 1}, {"lr", 2}, {"grad", 3}, {"m…
220 const OptimOriginIdx kSparseAdamOriginIdx = {{"weight", 0}, {"m", 1}, {"v", 2}, {"bet…
235 const OptimOriginIdx kSparseFtrlOriginIdx = {{"weight", 0}, {"accum", 1}, {"linear", 2}, {"grad", 3…
239 const std::map<std::string, OptimOriginIdx> kOptimToOriginIdx = {{kApplyMomentum, kMomentumOriginId…
243 const std::map<std::string, OptimOriginIdx> kOptimToPSSendIdx = {{kApplyMomentum, kMomentumPSSendId…
|
| D | optimizer_info_builder.cc | 68 const OptimOriginIdx &origin_input_map = kOptimToOriginIdx.at(optim_type); in GenInputAddrPtr()
|
| D | optimizer_info.cc | 52 const OptimOriginIdx &origin_input_map = kOptimToOriginIdx.at(optim_type); in UpdateOptimInputValue()
|