[Tosa] Fix 2d tfl batch_matmul legalization
Hi @rsuderman Can you please review this PR ? Thank you! Would you mind rebasing on head? Seems a merge conflict > Would you mind rebasing on head? Seems a merge conflict

done Hi @jpienaar  Can you please review this PR ? Thank you!
This patch fixes legalization of 2d tfl.batch_matmul to Tosa, which needs to be reshaped to 3d inputs because tosa matmul requires 3d inputs. And the 3d result needs to be reshaped back to original 2d.

added lit test for this as well.


Change-Id: I7f01cdebbdb6456d02a74f7f3b796ae0eccd1c8e