| /third_party/mindspore/mindspore/nn/optim/ |
| D | momentum.py | 28 …or_run_opt_ext(opt, momentum, learning_rate, gradient, weight, moment, ps_parameter, cache_enable): argument
30 if ps_parameter and not cache_enable:
171 … lr, gradients, params, moments, self.ps_parameters, self.cache_enable)
174 … gradients, params, moments, self.ps_parameters, self.cache_enable)
|
| D | lazyadam.py | 35 beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable): argument
40 if ps_parameter and not cache_enable:
82 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable): argument
85 if ps_parameter and not cache_enable:
275 self.cache_enable)
282 self.cache_enable)
|
| D | ftrl.py | 30 gradient, weight, moment, ps_parameter, cache_enable): argument
35 if ps_parameter and not cache_enable:
47 gradient, weight, moment, ps_parameter, cache_enable): argument
50 if ps_parameter and not cache_enable:
234 linear, grads, params, moments, self.ps_parameters, self.cache_enable)
|
| D | adam.py | 94 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
99 if ps_parameter and not cache_enable:
156 moment1, moment2, ps_parameter, cache_enable): argument
159 if ps_parameter and not cache_enable:
359 … lr, gradients, params, moment1, moment2, self.ps_parameters, self.cache_enable)
364 … gradients, params, moment1, moment2, self.ps_parameters, self.cache_enable)
|
| D | optimizer.py | 194 cache_filter = lambda x: x.cache_enable
195 self.cache_enable = tuple(cache_filter(x) for x in self.parameters)
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 136 self.cache_enable = False
290 if _is_role_worker() and self.cache_enable:
399 x.cache_enable = self.cache_enable
438 def cache_enable(self): member in Parameter
440 return self.param_info.cache_enable
442 @cache_enable.setter
443 def cache_enable(self, value=True): member in Parameter
446 self.param_info.cache_enable = value
576 if _is_role_worker() and self.cache_enable:
661 if not x1.cache_enable:
|
| /third_party/mindspore/mindspore/ccsrc/ps/ps_cache/ps_data/ |
| D | ps_data_prefetch.h | 37 EXPORT bool cache_enable() const { return cache_enable_; } in cache_enable() function
38 EXPORT void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable() argument
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | param_info.h | 91 bool cache_enable() const { return cache_enable_; } in cache_enable() function
92 void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable() argument
|
| D | tensor.h | 312 bool cache_enable() const { return cache_enable_; } in cache_enable() function
313 void set_cache_enable(bool cache_enable = true) { cache_enable_ = cache_enable; }
|
| /third_party/mindspore/mindspore/nn/layer/ |
| D | embedding.py | 233 self.cache_enable = self.vocab_cache_size > 0
259 if self.cache_enable and enable_ps:
279 … if (target == 'DEVICE' and not full_batch) or (self.cache_enable and enable_ps and sparse):
305 if self.cache_enable and not enable_ps:
331 self.embedding_table.cache_enable = self.cache_enable
337 self.cache_enable = False
361 self.cache_enable = True
373 self.embedding_table.cache_enable = True
|
| D | thor_layer.py | 729 self.cache_enable = self.vocab_cache_size > 0
758 if self.cache_enable and enable_ps:
778 … if (target == 'DEVICE' and not full_batch) or (self.cache_enable and enable_ps and sparse):
804 if self.cache_enable and not enable_ps:
861 self.embedding_table.cache_enable = self.cache_enable
867 self.cache_enable = False
891 self.cache_enable = True
901 self.embedding_table.cache_enable = True
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 221 def _set_cache_enable(cache_enable): argument
224 if cache_enable:
228 ps_context().set_cache_enable(cache_enable)
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime_manager.cc | 30 if (ps::PSContext::instance()->is_worker() && ps::PsDataPrefetch::GetInstance().cache_enable()) { in ClearRuntimeResource()
136 if (ps::PSContext::instance()->is_worker() && ps::PsDataPrefetch::GetInstance().cache_enable()) { in ReleaseKernelRuntime()
|
| /third_party/mindspore/tests/st/fl/mobile/src/ |
| D | adam.py | 109 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
114 if ps_parameter and not cache_enable:
177 moment1, moment2, ps_parameter, cache_enable): argument
180 if ps_parameter and not cache_enable:
|
| /third_party/mindspore/tests/st/fl/albert/src/ |
| D | adam.py | 110 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
115 if ps_parameter and not cache_enable:
177 moment1, moment2, ps_parameter, cache_enable): argument
180 if ps_parameter and not cache_enable:
|
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | adam.py | 109 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
114 if ps_parameter and not cache_enable:
176 moment1, moment2, ps_parameter, cache_enable): argument
179 if ps_parameter and not cache_enable:
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 92 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in Reset()
177 void PSContext::set_cache_enable(bool cache_enable) const { in set_cache_enable()
179 PsDataPrefetch::GetInstance().set_cache_enable(cache_enable); in set_cache_enable()
|
| D | ps_context.h | 81 void set_cache_enable(bool cache_enable) const;
|
| D | parameter_server.cc | 235 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in InitEmbeddingTable()
488 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in GetEmbeddingTableParamPtr()
740 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in HandleInitEmbeddings()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | param_info_py.cc | 38 … .def_property("cache_enable", &ParamInfo::cache_enable, &ParamInfo::set_cache_enable) in __anon2c0500310102()
|
| D | tensor_py.cc | 334 if (tensor.cache_enable()) { in FlushFromCache()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | unique_info.cc | 160 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in replace_graph()
|
| D | gather_v2_p_info.cc | 166 if (ps::PsDataPrefetch::GetInstance().cache_enable() && enable_sparse) { in GetAttrs()
641 if (ps::PsDataPrefetch::GetInstance().cache_enable()) { in InferBias()
|
| /third_party/mindspore/mindspore/train/callback/ |
| D | _checkpoint.py | 446 if param.cache_enable:
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 44 if (param_info && param_info->cache_enable()) { in AddCacheParameters()
114 if (param_info && param_info->cache_enable()) { in FindParamCacheEnable()
|