Fix issue with transposing shape in CMSIS-NN batch matmul #4495

Job	Run time
no-labels	1s
ci-ready-to-merge	1s
ci-run	3s
ci-run-full	6s
call-ci / Bazel (presubmit)	16m 36s
call-ci / Bazel TFLite Tools (presubmit)	1m 46s
call-ci / Bazel msan (presubmit)	13m 49s
call-ci / Bazel asan (presubmit)	13m 51s
call-ci / Code Style (presubmit)	1m 18s
call-ci / Cortex-M Bluepill No Release (presubmit)	8m 28s
call-ci / Cortex-M Bluepill Release (presubmit)	7m 2s
call-ci / Cortex-M Bluepill Renode (presubmit)	5m 24s
call-ci / Cortex-M QEMU Unit Tests (presubmit)	7m 13s
call-ci / Project Generation (presubmit)	9m 31s
call-ci / Makefile x86 Release (presubmit)	11m 49s
call-ci / Makefile x86 Default (presubmit)	6m 44s
call-ci / Makefile x86 No TFLite Static Memory (presubmit)	5m 50s
call-ci / Makefile x86 Out Of Tree (presubmit)	3m 38s
call-hexagon / Hexagon Build Test (presubmit)	8m 40s
call-xtensa-presubmit / Hifi3z Unit Tests (presubmit)	23m 12s
call-xtensa-presubmit / Hifi5 Unit Tests (presubmit)	19m 27s
call-xtensa-presubmit / Vision P6 Build (presubmit)	12m 8s
call-check-tflite-files / Check PR Modifies TfLite Files	37s
PR has Bug	1s
	2h 57m 15s

Provide feedback