Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_sparse.py1858 def _test_basic_ops_shape(self, nnz_x1, nnz_x2, shape_i, shape_v, dtype, device, coalesced): member in TestSparse
1938 self._test_basic_ops_shape(9, 12, [5, 6], [], dtype, device, coalesced)
1939 self._test_basic_ops_shape(9, 12, [10, 10, 10], [], dtype, device, coalesced)
1940 self._test_basic_ops_shape(9, 12, [50, 30, 20], [], dtype, device, coalesced)
1941 self._test_basic_ops_shape(9, 12, [5, 5, 5, 5, 5, 5], [], dtype, device, coalesced)
1942 self._test_basic_ops_shape(0, 12, [10, 10, 10], [], dtype, device, coalesced)
1943 self._test_basic_ops_shape(9, 0, [10, 10, 10], [], dtype, device, coalesced)
1944 self._test_basic_ops_shape(0, 0, [10, 10, 10], [], dtype, device, coalesced)
1945 self._test_basic_ops_shape(0, 0, [10, 10, 0], [], dtype, device, coalesced)
1946 self._test_basic_ops_shape(0, 0, [], [], dtype, device, coalesced)
[all …]