| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/distributed/ps/ps_cache/ |
| D | ps_data_prefetch.h | 34 EXPORT bool cache_enable() const { return cache_enable_; } in cache_enable() function
35 EXPORT void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ |
| D | ftrl.py | 32 gradient, weight, moment, cache_enable, argument
84 gradient, weight, moment, cache_enable, argument
103 gradient, weight, moment, cache_enable, argument
117 gradient, weight, moment, cache_enable): argument
129 gradient, weight, moment, cache_enable): argument
140 gradient, weight, moment, cache_enable): argument
|
| D | lazyadam.py | 39 … beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable, argument
93 … ps_parameter, cache_enable, distributed_opt, use_flag, distributed_sparse_opt, argument
133 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable, argument
154 beta1, beta2, eps, lr, gradient, params, m, v, ps_parameter, cache_enable): argument
203 cache_enable): argument
235 … beta1, beta2, eps, lr, gradient, params, moment1, moment2, ps_parameter, cache_enable): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ir/ |
| D | param_info.h | 112 bool cache_enable() const { return cache_enable_; } in cache_enable() function
113 void set_cache_enable(bool cache_enable) { cache_enable_ = cache_enable; } in set_cache_enable()
|
| D | tensor.h | 372 bool cache_enable() const { return cache_enable_; } in cache_enable() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/ |
| D | _ps_context.py | 195 def _set_cache_enable(cache_enable): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/common/ |
| D | parameter.py | 670 def cache_enable(self): member in Parameter
685 def cache_enable(self, value=True): member in Parameter
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/ps/ |
| D | ps_context.cc | 218 bool PSContext::cache_enable() const { in cache_enable() function in mindspore::ps::PSContext
|