Searched refs:_get_gradients_mean (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/tests/st/fl/hybrid_lenet/src/ |
| D | cell_wrapper.py | 17 from mindspore.parallel._utils import (_get_device_num, _get_gradients_mean,
93 mean = _get_gradients_mean()
|
| /third_party/mindspore/mindspore/nn/wrap/ |
| D | cell_wrapper.py | 18 from mindspore.parallel._utils import (_get_device_num, _get_gradients_mean,
348 self.mean = _get_gradients_mean()
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 25 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean
297 mean = _get_gradients_mean()
|
| /third_party/mindspore/mindspore/boost/ |
| D | boost_cell_wrapper.py | 19 from mindspore.parallel._utils import _get_global_rank, _get_device_num, _get_gradients_mean
159 self.mean = _get_gradients_mean()
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 30 from mindspore.parallel._utils import _get_device_num, _get_parallel_mode, _get_gradients_mean
312 mean = _get_gradients_mean()
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _utils.py | 172 def _get_gradients_mean(): function
|
| /third_party/mindspore/tests/st/networks/models/resnet50/src_thor/ |
| D | thor.py | 24 from mindspore.parallel._utils import _get_device_num, _get_gradients_mean
238 mean = _get_gradients_mean()
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | thor.py | 26 from mindspore.parallel._utils import _get_device_num, _get_gradients_mean
449 mean = _get_gradients_mean()
767 mean = _get_gradients_mean()
|