(self)
| 107 | return self._to_backend("gpu") |
| 108 | |
| 109 | def cpu(self) -> DataNode: |
| 110 | self._check_gpu2cpu() |
| 111 | return self._to_backend("cpu") |
| 112 | |
| 113 | # Note: Regardless of whether we want the cpu or gpu version |
| 114 | # of a tensor, we keep the source argument the same so that |