Home
last modified time | relevance | path

Searched refs:kOptimToPSSendIdx (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info_builder.cc65 if (kOptimToOriginIdx.count(optim_type) == 0 || kOptimToPSSendIdx.count(optim_type) == 0) { in GenInputAddrPtr()
69 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in GenInputAddrPtr()
Doptimizer_info.cc49 if (kOptimToOriginIdx.count(optim_type) == 0 || kOptimToPSSendIdx.count(optim_type) == 0) { in UpdateOptimInputValue()
53 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in UpdateOptimInputValue()
Dconstants.h243 const std::map<std::string, OptimOriginIdx> kOptimToPSSendIdx = {{kApplyMomentum, kMomentumPSSendId… variable