torch_xla.xla_shardingas

Creator

Created

2024 Feb 26 7:48

Editor

Edited

2024 Feb 26 7:53

Refs


import torch_xla.experimental.xla_shardingas xs
from torch_xla.experimental.xla_shardingimportMesh

////////