Searched refs:optim_type (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 56 AddressPtr OptimizerInfoBuilder::GenInputAddrPtr(const std::string &optim_type, const std::string &… in GenInputAddrPtr() argument
61 …MS_LOG(INFO) << "Get input address pointer for optimizer:" << optim_type << ", input name:" << inp… in GenInputAddrPtr()
65 if (kOptimToOriginIdx.count(optim_type) == 0 || kOptimToPSSendIdx.count(optim_type) == 0) { in GenInputAddrPtr()
66 MS_LOG(EXCEPTION) << "Optimizer type " << optim_type << " in not supported."; in GenInputAddrPtr()
68 const OptimOriginIdx &origin_input_map = kOptimToOriginIdx.at(optim_type); in GenInputAddrPtr()
69 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in GenInputAddrPtr()
71 MS_LOG(EXCEPTION) << "Optimizer " << optim_type << " has no input for " << input_name; in GenInputAddrPtr()
|
| D | optimizer_info.cc | 46 void OptimizerInfo::UpdateOptimInputValue(const std::string &optim_type, const std::string &input_n… in UpdateOptimInputValue() argument
49 if (kOptimToOriginIdx.count(optim_type) == 0 || kOptimToPSSendIdx.count(optim_type) == 0) { in UpdateOptimInputValue()
50 MS_LOG(EXCEPTION) << "Optimizer type " << optim_type << " in not supported."; in UpdateOptimInputValue()
52 const OptimOriginIdx &origin_input_map = kOptimToOriginIdx.at(optim_type); in UpdateOptimInputValue()
53 const OptimPSSendIdx &ps_send_index_map = kOptimToPSSendIdx.at(optim_type); in UpdateOptimInputValue()
55 MS_LOG(EXCEPTION) << "Optimizer " << optim_type << " has no input for " << input_name; in UpdateOptimInputValue()
61 …MS_LOG(EXCEPTION) << "Index is out of bound for optimizer " << optim_type << ", origin_index:" << … in UpdateOptimInputValue()
|
| D | optimizer_info_builder.h | 49 …AddressPtr GenInputAddrPtr(const std::string &optim_type, const std::string &input_name, void *ps_…
|
| D | optimizer_info.h | 53 …void UpdateOptimInputValue(const std::string &optim_type, const std::string &input_name, void *dat…
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | other_ops.py | 692 def __init__(self, optim_type='ApplyMomentum', only_shape_indices=None): argument
|