PR #12975: [XLA:CPU][oneDNN] Fix failures in onednn_matmul_tests
PR #12975: [XLA:CPU][oneDNN] Fix failures in onednn_matmul_tests

Imported from GitHub PR https://github.com/openxla/xla/pull/12975

This PR fixes failures seen in onednn_matmul_tests
Copybara import of the project:

--
6ff6bd59a20fdabaf450134d6b449348a7a3c2ed by Kanvi Khanna <kanvi.khanna@intel.com>:

Fix failures in onednn_matmul_tests

Merging this change closes #12975

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/12975 from Intel-tensorflow:kanvi/fix_mm_tests 6ff6bd59a20fdabaf450134d6b449348a7a3c2ed
