(self, a, type_as)
| 1529 | return a.device_buffer.device() |
| 1530 | |
| 1531 | def _change_device(self, a, type_as): |
| 1532 | return jax.device_put(a, self._get_device(type_as)) |
| 1533 | |
| 1534 | def _from_numpy(self, a, type_as=None): |
| 1535 | if isinstance(a, float): |
no test coverage detected