(x, devices, indices, sharding=None)
| 357 | |
| 358 | |
| 359 | def _shard_mge_tensor(x, devices, indices, sharding=None): |
| 360 | x_np = x.numpy() |
| 361 | if x_np.shape == (): |
| 362 | x_np = np.array([x_np]) |
| 363 | return device_put([x_np[i] for i in indices], devices) |
| 364 | |
| 365 | |
| 366 | for nt in _np_types: |
nothing calls this directly
no test coverage detected