Searched defs:_support_parallel_optimizer (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/optim/ | ||
| D | lamb.py | 252 _support_parallel_optimizer = True variable in Lamb |
| D | adafactor.py | 279 _support_parallel_optimizer = True variable in AdaFactor |
| D | optimizer.py | 198 _support_parallel_optimizer = False variable in Optimizer |
| D | adam.py | 1014 _support_parallel_optimizer = True variable in AdamWeightDecay |
| /third_party/mindspore/mindspore-src/source/tests/st/optimizer/ | ||
| D | came.py | 235 _support_parallel_optimizer = True variable in Came |