Searched refs:test_dict_tensor_key (Results 1 – 1 of 1) sorted by relevance
65 def test_dict_tensor_key(a, t): function75 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))