Searched defs:compute_output (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/wrap/ |
| D | loss_scale.py | 471 def _check_overflow_status_on_infnan_mode(self, grad_overflow_check_func, compute_output): argument
479 …ef _get_distributed_overflow_status_on_infnan_mode(self, grad_overflow_check_func, compute_output): argument
491 def _get_distributed_overflow_status_on_infnan_enable_allfinite(self, compute_output): argument
500 def _get_gpu_overflow_status(self, compute_output): argument
505 def _get_ascend_overflow_status_on_infnan_mode(self, compute_output): argument
514 def _get_ascend_overflow_status_on_saturation_mode(self, status, compute_output): argument
539 def get_overflow_status(self, status, compute_output): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_npu_overflow_v2.py | 47 def get_overflow_status_v1(self, status, compute_output): argument
64 def get_overflow_status_v2(self, status, compute_output): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/boost/ |
| D | boost_cell_wrapper.py | 649 def _get_overflow_status(self, status, compute_output): argument
|