Searched refs:_test_resize_shape (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_sparse.py | 3032 def _test_resize_shape(self, x_i, x_v, x_size, y_i, y_v, y_size, dtype, device): member in TestSparse
3056 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
3060 self._test_resize_shape([1, 1], [1, 2, 0], [2, 2, 0],
3065 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
3070 self._test_resize_shape([1, 0], [0, 2, 3], [2, 2, 3],
3074 self._test_resize_shape([1, 0], [0, 2, 3], [2, 2, 3],
3080 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
3085 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
3091 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
3097 self._test_resize_shape([1, 1], [1, 2, 3], [2, 2, 3],
[all …]
|