| /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()
|
| /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()
|
| D | tensor.h | 312 bool cache_enable() const { return cache_enable_; } in cache_enable() function
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | ftrl.py | 30 gradient, weight, moment, ps_parameter, cache_enable): argument
47 gradient, weight, moment, ps_parameter, cache_enable): argument
|
| D | lazyadam.py | 35 beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable): argument
82 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable): argument
|
| D | momentum.py | 28 …or_run_opt_ext(opt, momentum, learning_rate, gradient, weight, moment, ps_parameter, cache_enable): argument
|
| D | adam.py | 94 … beta2_power, beta1, beta2, eps, lr, gradient, param, m, v, ps_parameter, cache_enable): argument
156 moment1, moment2, ps_parameter, cache_enable): argument
|
| /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
177 moment1, moment2, ps_parameter, cache_enable): argument
|
| /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
177 moment1, moment2, ps_parameter, cache_enable): argument
|
| /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
176 moment1, moment2, ps_parameter, cache_enable): argument
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _ps_context.py | 221 def _set_cache_enable(cache_enable): argument
|
| /third_party/mindspore/mindspore/common/ |
| D | parameter.py | 438 def cache_enable(self): member in Parameter
443 def cache_enable(self, value=True): member in Parameter
|