PR #10759: [XLA:CPU][oneDNN] Enable matrix-vector and vector-vector product
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA).

View this [failed invocation](https://github.com/tensorflow/tensorflow/pull/64604/checks?check_run_id=23166609496) of the CLA check for more information.

For the most up to date status, view the checks section at the bottom of the pull request.
PR #10759: [XLA:CPU][oneDNN] Enable matrix-vector and vector-vector product

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

This PR relaxes conditions to rewrite dot operations of the form vector-matrix, matrix-vector, or vector-vector to oneDNN custom calls, provided the original problem meets the empirically determined multiply-accumulate threshold. In particular this PR:

1. Relaxes some constraints on Dot to oneDNN matmul custom call conversion
2. Reconfigures the dimensions of the operands and outputs of convertible dot operations.
3. Adds tests to verify rewrite and execution result
Copybara import of the project:

--
d24e5cd0b77d0734a0f33011ae03127f00d80e7d by Akhil Goel <akhil.goel@intel.com>:

Relax constraints for matmul rewrite

--
9582288fc5d06eb8f4641be32d6c41746f13dbce by Akhil Goel <akhil.goel@intel.com>:

Fix gemv test after merge

--
604d4fbd29a78a7d305092356176e31081ea4ff0 by Akhil Goel <akhil.goel@intel.com>:

Address review comments

--
4534fbd7e96a6e45ccc5a501b38b940aa8bd9d38 by Akhil Goel <akhil.goel@intel.com>:

Optional commit

Merging this change closes #10759

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10759 from Intel-tensorflow:akhil/gemv 4534fbd7e96a6e45ccc5a501b38b940aa8bd9d38
