Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dutils.py1354 def _setup_trace_module_map( function
1537 module_typenames_to_export_as_functions = _setup_trace_module_map(
H A Dverification.py599 utils._setup_trace_module_map(model, export_modules_as_functions)
/aosp_15_r20/external/pytorch/test/onnx/
H A Dtest_utility_funs.py70 torch.onnx.utils._setup_trace_module_map(model, False)