Searched refs:_test_extract_gradients_from_optimizer (Results 1 – 1 of 1) sorted by relevance
220 def _test_extract_gradients_from_optimizer(self, set_to_none: bool) -> None: member in TestIdentifyGradients273 self._test_extract_gradients_from_optimizer(set_to_none=False)276 self._test_extract_gradients_from_optimizer(set_to_none=True)