Home
last modified time | relevance | path

Searched refs:_checkScriptable (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_quantization.py713 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 Dtest_quantize_eager_ptq.py1388 self._checkScriptable(model_with_packed_input, scripted, [[packed_input]], True)