Searched refs:_test_setitem_tensor (Results 1 – 1 of 1) sorted by relevance
/aosp_15_r20/external/pytorch/test/ |
H A D | test_autograd.py | 3454 def _test_setitem_tensor(self, size, index): member in TestAutograd 3486 self._test_setitem_tensor((5, 5), 3) 3487 self._test_setitem_tensor((5, 5), [[0, 1], [1, 0]]) 3488 self._test_setitem_tensor((5,), 3) 3489 self._test_setitem_tensor( 3492 self._test_setitem_tensor((5,), [[0, 1, 2, 3]]) 3493 self._test_setitem_tensor((5, 5, 5), [slice(None), slice(None), [1, 3]]) 3494 self._test_setitem_tensor((5, 5, 5), [slice(None), [1, 3], slice(None)]) 3495 self._test_setitem_tensor((5, 5, 5), [[1, 3], slice(None), slice(None)]) 3496 self._test_setitem_tensor((5, 5, 5), [slice(None), [2, 4], [1, 3]]) [all …]
|