Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dlogging_tensor.py86 class LoggingTensorMode(TorchDispatchMode): class
181 with LoggingTensorMode(), capture_logs(True, python_tb, script_tb, cpp_tb) as logs:
/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py38 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()
H A Dtest_proxy_tensor.py288 from torch.testing._internal.logging_tensor import LoggingTensorMode
291 with LoggingTensorMode():
297 with LoggingTensorMode(), LoggingTensorMode():
/aosp_15_r20/external/pytorch/torch/utils/
H A Dcheckpoint.py17 from torch.testing._internal.logging_tensor import capture_logs, LoggingTensorMode
990 with LoggingTensorMode(), \