Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py362 def _test_unbind_case(self, a, b): member in TestNestedTensor
379 self._test_unbind_case(torch.tensor([1, 2]), torch.tensor([7, 8]))
383 self._test_unbind_case(torch.tensor([1]), torch.tensor([7]))
387 self._test_unbind_case(torch.tensor([1.0]), torch.tensor([]))
391 self._test_unbind_case(torch.tensor([]), torch.tensor([]))