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()
|