Searched refs:get_script_module (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/test/quantization/jit/ |
| H A D | test_ondevice_quantization.py | 18 get_script_module,
60 scripted_model = get_script_module(model, False, inputs)
67 m = get_script_module(model, False, inputs)
141 orig_scripted_model = get_script_module(model, False, inputs)
|
| H A D | test_quantize_jit.py | 61 get_script_module,
154 scripted_or_traced = get_script_module(eager, tracing, x).eval()
198 scripted_or_traced = get_script_module(eager, tracing, x).eval()
248 scripted_or_traced = get_script_module(eager, tracing, x).eval()
293 scripted_or_traced = get_script_module(eager, tracing, x)
366 scripted_or_traced = get_script_module(eager, tracing, x).eval()
528 script_model = get_script_module(eager_model, tracing, x)
1302 model = get_script_module(
3314 model = get_script_module(eager_model, tracing, x)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_quantization.py | 441 def get_script_module(model, tracing, data): function
751 model = get_script_module(module, tracing, inputs[0]).eval()
|