Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/quantization/jit/
H A Dtest_ondevice_quantization.py18 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 Dtest_quantize_jit.py61 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 Dcommon_quantization.py441 def get_script_module(model, tracing, data): function
751 model = get_script_module(module, tracing, inputs[0]).eval()