Home
last modified time | relevance | path

Searched refs:checkScriptable (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py311 self.checkScriptable(model, [[x]], check_save_load=True)
334 self.checkScriptable(model, [[xq]], check_save_load=True)
361 self.checkScriptable(model, self.calib_data)
405 self.checkScriptable(model, self.calib_data)
447 self.checkScriptable(model, self.calib_data)
488 self.checkScriptable(model, self.calib_data)
530 self.checkScriptable(model, self.calib_data)
559 self.checkScriptable(model, self.calib_data)
585 self.checkScriptable(model, self.calib_data)
645 self.checkScriptable(model, self.calib_data)
[all …]
H A Dtest_quantize_eager_qat.py277 self.checkScriptable(model, self.calib_data)
299 self.checkScriptable(model, self.calib_data)
339 self.checkScriptable(model, self.img_data_2d)
370 self.checkScriptable(model, self.img_data_2d)
400 self.checkScriptable(model, self.calib_data)
428 self.checkScriptable(model, self.embed_data)
456 self.checkScriptable(model, self.embed_data)
/aosp_15_r20/external/pytorch/test/quantization/core/
H A Dtest_quantized_module.py128 self.checkScriptable(qlinear_copy, [[X_q]], check_save_load=True)
177 self.checkScriptable(copy.deepcopy(loaded_qlinear), [[X_q]], check_save_load=True)
230 self.checkScriptable(qlinear, [[X_q]], check_save_load=True)
399 self.checkScriptable(
919 self.checkScriptable(pool_under_test, [[X]])
1514 self.checkScriptable(
1671 self.checkScriptable(qlinear, [[X]], check_save_load=True)
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py711 def checkScriptable(self, orig_mod, calib_data, check_save_load=False): member in QuantizationTestCase
1151 self.checkScriptable(qemb, [inputs], check_save_load=True)
/aosp_15_r20/external/pytorch/test/quantization/fx/
H A Dtest_quantize_fx.py8734 self.checkScriptable(model_graph, [[sample_input]], True)
9574 self.checkScriptable(model, eval_output)
9614 self.checkScriptable(model, eval_output)