Home
last modified time | relevance | path

Searched refs:StopPrint (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/tests/ut/python/pynative_mode/
Dtest_stop_gradient.py435 class StopPrint(nn.Cell): class
437 super(StopPrint, self).__init__()
445 grad_all(StopPrint())(Tensor(np.ones([2]).astype(np.float32)),