Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_save_load_for_op_version.py247 class MyModuleFloat(torch.nn.Module): class
267 current_mobile_module_float = self._save_load_mobile_module(MyModuleFloat)
309 class MyModuleFloat(torch.nn.Module): class
329 current_mobile_module_float = self._save_load_mobile_module(MyModuleFloat)
381 class MyModuleFloat(torch.nn.Module): class
403 current_mobile_module_float = self._save_load_module(MyModuleFloat)
/aosp_15_r20/external/pytorch/torch/csrc/jit/operator_upgraders/
H A DREADME.md165 class MyModuleFloat(torch.nn.Module):
181 current_mobile_module_float = self._save_load_mobile_module(MyModuleFloat)
182 current_server_module_float = self._save_load_module(MyModuleFloat)