Update __init__.py

pytorch · Dec 4, 2023 · b49b2dd · b49b2dd
1 parent 58a7763
commit b49b2dd
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torch_xla/distributed/spmd/__init__.py b/torch_xla/distributed/spmd/__init__.py
@@ -3,7 +3,6 @@
                            XLAPatchedLinear, mark_sharding, clear_sharding,
                            wrap_if_sharded, xla_patched_nn_linear_forward)
 from .api import xla_distribute_tensor, xla_distribute_module
-from .debugging import visualize_tensor_sharding
 
 __all__ = [
     "XLAShard",