Searched refs:assertObjectIn (Results 1 – 5 of 5) sorted by relevance
/aosp_15_r20/external/pytorch/test/quantization/eager/ |
H A D | test_quantize_eager_ptq.py | 748 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 D | test_quantize_eager_qat.py | 543 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 D | test_hub.py | 110 self.assertObjectIn("mnist", entry_lists)
|
H A D | test_nn.py | 2421 self.assertObjectIn(new_param, l.parameters())
|
/aosp_15_r20/external/pytorch/torch/testing/_internal/ |
H A D | common_utils.py | 3903 def assertObjectIn(self, obj: Any, iterable: Iterable[Any]) -> None: member in TestCase
|