Searched refs:_setup_trace_module_map (Results 1 – 3 of 3) sorted by relevance
1354 def _setup_trace_module_map( function1537 module_typenames_to_export_as_functions = _setup_trace_module_map(
599 utils._setup_trace_module_map(model, export_modules_as_functions)
70 torch.onnx.utils._setup_trace_module_map(model, False)