Home
last modified time | relevance | path

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

/third_party/mindspore/tests/ut/python/parallel/
Dtest_shared_param_and_mix_precision.py61 gradient_fp32_sync=True): argument
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.h72 bool gradient_fp32_sync() const { return gradient_fp32_sync_; } in gradient_fp32_sync() function
Dcontext.cc102 void ParallelContext::set_gradient_fp32_sync(bool gradient_fp32_sync) { gradient_fp32_sync_ = gradi… in set_gradient_fp32_sync()
/third_party/mindspore/mindspore/parallel/
D_auto_parallel_context.py139 def set_gradient_fp32_sync(self, gradient_fp32_sync): argument