Searched refs:assertGradientDetected (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_memory_profiler.py | 137 def assertGradientDetected(self, name: str, *args, **kwargs) -> None: member in TestIdentifyGradients
170 self.assertGradientDetected("w0", prof, _EventType.TorchOp, w0.grad)
171 self.assertGradientDetected("w1", prof, _EventType.TorchOp, w1.grad)
203 self.assertGradientDetected(name, prof, _EventType.TorchOp, p.grad)
211 self.assertGradientDetected(name, prof, _EventType.PyCall, p.grad, p)
237 self.assertGradientDetected("w0", prof, _EventType.PyCall, w0.grad, w0)
238 self.assertGradientDetected("w1", prof, _EventType.PyCall, w1.grad, w1)
240 self.assertGradientDetected("w0", prof, _EventType.TorchOp, w0.grad)
241 self.assertGradientDetected("w1", prof, _EventType.TorchOp, w1.grad)
288 self.assertGradientDetected(
|