-
Smit Hinsu authored
TPUs don't support dynamic Squeeze as it won't be able to identify the dimensions to remove if the input shape is dynamic. For example, [<=16, 1] could return a vector or scalar result depending on the size of the dynamic dim. PiperOrigin-RevId: 415083667
ce7d8065
