Fix conditions for `tfl.batch_matmul` legalization and e2e tests.
Fix conditions for `tfl.batch_matmul` legalization and e2e tests.

This CL also:
* Adds TFLite op check to `QuantizationConfig` preset tests in ODML.
