Removed named_shape= from jax.core.ShapedArray and `jax.ShapeDtyp…

…eStruct` It is unused and was only kept around to avoid breaking internal users. PiperOrigin-RevId: 673311219
google-deepmind · Sep 11, 2024 · bae3994 · bae3994
1 parent f19ab86
commit bae3994
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/treescope/external/jax_support.py b/treescope/external/jax_support.py
@@ -304,8 +304,6 @@ def render_shape_dtype_struct(
       "shape": node.shape,
       "dtype": node.dtype,
   }
-  if node.named_shape:
-    attributes["named_shape"] = node.named_shape
   if node.sharding is not None:
     attributes["sharding"] = node.sharding