Searched refs:python_tb (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | logging_tensor.py | 153 def capture_logs(is_mode=False, python_tb=False, script_tb=False, cpp_tb=False) -> Iterator[List[st… argument
154 collect_traceback = python_tb or script_tb or cpp_tb
167 logger.addFilter(GatherTraceback(python=python_tb, script=script_tb, cpp=cpp_tb))
180 def capture_logs_with_logging_tensor_mode(python_tb=False, script_tb=False, cpp_tb=False): argument
181 with LoggingTensorMode(), capture_logs(True, python_tb, script_tb, cpp_tb) as logs:
|
| /aosp_15_r20/external/pytorch/torch/utils/ |
| H A D | checkpoint.py | 991 … capture_logs(True, python_tb=True, script_tb=True, cpp_tb=cpp_tb) as logs_and_tb:
|