[Resubmit][TF2XLA] Use dynamic dimension in SizeOp.
This is needed if the number of elements in a tensor can be dynamic. Resubmit after xla_builder support constant-folding get_dynamic_size. PiperOrigin-RevId: 228974238
Loading
Please sign in to comment