Home
last modified time | relevance | path

Searched defs:gradient_fp32_sync (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_shared_param_and_mix_precision.py67 gradient_fp32_sync=True, search_mode="dynamic_programming"): argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/utils/
Dparallel_context.cc153 void ParallelContext::set_gradient_fp32_sync(bool gradient_fp32_sync) { gradient_fp32_sync_ = gradi… in set_gradient_fp32_sync()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/common/utils/
Dparallel_context.h85 bool gradient_fp32_sync() const { return gradient_fp32_sync_; } in gradient_fp32_sync() function
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_auto_parallel_context.py321 def set_gradient_fp32_sync(self, gradient_fp32_sync): argument