Fix: input and output must have same shape for matching shapes (#1357) #363

Job	Run time
build-and-test / build-image	9s
pre-commit / pre-commit	40s
macos-build / build (macos-latest, clang, clang++, Release, OFF)	1m 58s
spdx / check-spdx-headers	4s
build-and-test / Build and test tt-explorer (n150, ON, perf)	16m 37s
build-and-test / Build tt-mlir (ubuntu-latest, OFF, run)	34m 39s
build-and-test / Build tt-mlir (ubuntu-latest, ON, perf)	18m 6s
build-and-test / run-ttrt-tests (n150, ON, perf)	10m 0s
build-and-test / run-tests (n150, OFF, run, --non-zero)	4m 43s
build-and-test / run-tests (n150, ON, perf)	7m 13s
build-and-test / run-tests (n300, OFF, run, --non-zero)	4m 55s
build-and-test / run-tests (n300, ON, perf)	10m 40s
	1h 49m 44s

Provide feedback