Searched refs:test_all_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 6929 def test_all_tensor(x): function
6931 self.assertFalse(test_all_tensor(torch.tensor([1, 0, 3], dtype=torch.uint8)))
6932 self.assertTrue(test_all_tensor(torch.tensor([3.14, 3, 99], dtype=torch.uint8)))
6933 self.assertTrue(test_all_tensor(torch.tensor([True, True], dtype=torch.uint8)))
6934 self.assertFalse(test_all_tensor(torch.tensor([True, False], dtype=torch.uint8)))
|