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