Searched refs:LoggingTensorMode (Results 1 – 4 of 4) sorted by relevance
86 class LoggingTensorMode(TorchDispatchMode): class181 with LoggingTensorMode(), capture_logs(True, python_tb, script_tb, cpp_tb) as logs:
38 LoggingTensorMode,1377 with LoggingTensorMode():1389 with LoggingTensorMode():1399 with LoggingTensorMode():1400 with LoggingTensorMode():1516 with LoggingTensorMode():1521 with LoggingTensorMode():1696 with LoggingTensorMode() as reenabled:1748 x = LoggingTensorMode()1749 y = LoggingTensorMode()
288 from torch.testing._internal.logging_tensor import LoggingTensorMode291 with LoggingTensorMode():297 with LoggingTensorMode(), LoggingTensorMode():
17 from torch.testing._internal.logging_tensor import capture_logs, LoggingTensorMode990 with LoggingTensorMode(), \