Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_subclasses.py27 NestedTensorTestCase,
1806 class TestNestedTensor(torch._dynamo.test_case.TestCase, NestedTensorTestCase):
/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py52 NestedTensorTestCase,
293 class TestNestedTensor(NestedTensorTestCase):
849 class TestNestedTensorDeviceType(NestedTensorTestCase):
2749 class TestNestedTensorAutograd(NestedTensorTestCase):
3521 class TestNestedTensorSubclass(NestedTensorTestCase):
7294 class TestNestedTensorOpInfo(NestedTensorTestCase):
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_utils.py5255 class NestedTensorTestCase(TestCase): class