Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/ps/
Dconstants.h212 using OptimPSSendIdx = std::map<std::string, size_t>; variable
217 const OptimPSSendIdx kMomentumPSSendIdx = {
223 const OptimPSSendIdx kSparseAdamPSSendIdx = {{"weight", INDEX_NOT_SEND},
236 const OptimPSSendIdx kSparseFtrlPSSendIdx = {
Doptimizer_info_builder.cc69 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in GenInputAddrPtr()
Doptimizer_info.cc53 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in UpdateOptimInputValue()