Home
last modified time | relevance | path

Searched defs:set_gradient_fp32_sync (Results 1 – 2 of 2) sorted by relevance

/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() function in mindspore::parallel::ParallelContext
/third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/parallel/
D_auto_parallel_context.py321 def set_gradient_fp32_sync(self, gradient_fp32_sync): member in _AutoParallelContext