Searched refs:checkScriptable (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_quantize_eager_ptq.py | 311 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 D | test_quantize_eager_qat.py | 277 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 D | test_quantized_module.py | 128 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 D | common_quantization.py | 711 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 D | test_quantize_fx.py | 8734 self.checkScriptable(model_graph, [[sample_input]], True) 9574 self.checkScriptable(model, eval_output) 9614 self.checkScriptable(model, eval_output)
|