Searched refs:_test_basic_ops_shape (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_sparse.py | 1858 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 …]
|