Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/ao/quantization/
H A Dquantize_jit.py6 from torch.jit._recursive import wrap_cpp_module
64 model = wrap_cpp_module(model_c)
81 model = wrap_cpp_module(model_c)
105 model = wrap_cpp_module(model_c)
150 model = wrap_cpp_module(model_c)
178 model = wrap_cpp_module(model_c)
/aosp_15_r20/external/pytorch/torch/jit/
H A D_serialization.py17 from torch.jit._recursive import wrap_cpp_module
170 ret = wrap_cpp_module(cpp_module)
193 return wrap_cpp_module(torch._C._load_jit_module_from_file(f))
195 return wrap_cpp_module(torch._C._load_jit_module_from_bytes(f.read()))
H A D_script.py34 wrap_cpp_module,
397 return wrap_cpp_module(cpp_module)
687 modules[name] = wrap_cpp_module(cpp_module)
856 return torch.jit._recursive.wrap_cpp_module(copy.copy(self._c))
859 return torch.jit._recursive.wrap_cpp_module(copy.deepcopy(self._c, memo))
H A D_script.pyi25 wrap_cpp_module as wrap_cpp_module,
H A D_recursive.py1010 def wrap_cpp_module(cpp_module): function
1015 setattr(script_module, name, wrap_cpp_module(cpp_module))
/aosp_15_r20/external/pytorch/torch/utils/
H A Dbundled_inputs.py7 from torch.jit._recursive import wrap_cpp_module
119 cloned_module = wrap_cpp_module(clone)
H A Dmobile_optimizer.py76 return torch.jit._recursive.wrap_cpp_module(optimized_cpp_module)
/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_quantize_jit.py46 from torch.jit._recursive import wrap_cpp_module
1150 m = torch.jit._recursive.wrap_cpp_module(m._c)
2864 m = wrap_cpp_module(
2960 m = wrap_cpp_module(
3274 wt_module = wrap_cpp_module(qconfig.weight)
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_freezing.py11 from torch.jit._recursive import wrap_cpp_module
1486 m_frozen = wrap_cpp_module(torch._C._freeze_module(m._c))
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py11124 from torch.jit._recursive import wrap_cpp_module
11125 m = wrap_cpp_module(torch._C._freeze_module(m._c))