Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dlogging_tensor.py180 def capture_logs_with_logging_tensor_mode(python_tb=False, script_tb=False, cpp_tb=False): function
/aosp_15_r20/external/pytorch/test/
H A Dtest_python_dispatch.py35 capture_logs_with_logging_tensor_mode,
1416 with capture_logs_with_logging_tensor_mode() as logs:
1429 with capture_logs_with_logging_tensor_mode() as logs1:
1430 with capture_logs_with_logging_tensor_mode() as logs2:
H A Dtest_nestedtensor.py5893 capture_logs_with_logging_tensor_mode,
5900 with capture_logs_with_logging_tensor_mode():