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 …]
|