Searched refs:_checkScriptable (Results 1 – 2 of 2) sorted by relevance
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_quantization.py | 713 self._checkScriptable(orig_mod, scripted, calib_data, check_save_load) 717 self._checkScriptable(orig_mod, traced, calib_data, check_save_load) 720 def _checkScriptable(self, orig_mod, script_mod, calib_data, check_save_load): member in QuantizationTestCase
|
/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_quantize_eager_ptq.py | 1388 self._checkScriptable(model_with_packed_input, scripted, [[packed_input]], True)
|