Fix check whether there is more than one tile.
The previous check was checking the number of elements in a tile against the number of elements in the input shape. This doesn't work if one dimension of the tile is bigger than the input dimension, but the other dimension is smaller. PiperOrigin-RevId: 202326635
Loading
Please sign in to comment