Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_typing.py65 def test_dict_tensor_key(a, t): function
75 self.checkScript(test_dict_tensor_key, (dict_a, torch.tensor(4)))
76 self.checkScript(test_dict_tensor_key, (dict_a, torch.tensor(3)))
77 self.checkScript(test_dict_tensor_key, (dict_a, inp1))
78 self.checkScript(test_dict_tensor_key, (dict_a, inp2))