Fix convolution bug when input and filter dimensions match

TFLite has an optimized matmul path for cases where the input and
filter tensors have matching width+height. However, this case doesn't
properly account for multiple *batches*. Account for this and add
an appropriate test.

Credit to zgxnet for the bug and proposed fix.

Fixes #21817

PiperOrigin-RevId: 212645329