Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/onnx/
H A Dutils.py110 def disable_apex_o2_state_dict_hook(model: torch.nn.Module | torch.jit.ScriptFunction): function
156 ) as mode_ctx, disable_apex_o2_state_dict_hook(