Add "clear_output_shapes" option to FoldConstants transformer in
tools/graph_transforms. By setting this option to false, the transformer will not strip off the shape information stored as attributes. PiperOrigin-RevId: 172057283
Loading
Please sign in to comment