Remove failing assertion
Mixing v1/v2 TensorShapes seems fine except for this one assertion. I've verified that the object-based checkpointing tests pass with enable_v2_behavior() now (while failing on this assertion before). It was never an issue when setting the environmental variable, since everything saw a consistent TensorShape class. PiperOrigin-RevId: 226203984
Loading
Please sign in to comment