Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/quantization/eager/
H A Dtest_quantize_eager_ptq.py748 self.assertObjectIn(fw_pre_hook, model.fc._forward_pre_hooks.values())
749 self.assertObjectIn(fw_hook, model.fc._forward_hooks.values())
1465 self.assertObjectIn(fw_pre_hook, model.fc1._forward_pre_hooks.values())
1466 self.assertObjectIn(fw_hook, model.fc1._forward_hooks.values())
H A Dtest_quantize_eager_qat.py543 self.assertObjectIn(fw_pre_hook, model.fc._forward_pre_hooks.values())
544 self.assertObjectIn(fw_hook, model.fc._forward_hooks.values())
/aosp_15_r20/external/pytorch/test/
H A Dtest_hub.py110 self.assertObjectIn("mnist", entry_lists)
H A Dtest_nn.py2421 self.assertObjectIn(new_param, l.parameters())
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py3903 def assertObjectIn(self, obj: Any, iterable: Iterable[Any]) -> None: member in TestCase