PR #10211: [XLA:CPU][oneDNN] Update oneDNN LayerNorm and Softmax
PR #10211: [XLA:CPU][oneDNN] Update oneDNN LayerNorm and Softmax

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

This PR addresses following : 
1) Add support for FP16 to oneDNN softmax pattern matching
2) Adds another pattern for oneDNN layernorm based on keras layernorm operation
3) Fixes previous oneDNN layernorm pattern, as it was focused on flax layernorm pattern, which recently wen through an update while moment calculation
4) Adds FP16 support to oneDNN layernorm pattern matching.
Copybara import of the project:

--
9eff5f704d99a2a594226e6a9cc1db72a67c8de8 by Sachin Muradi <sachin.muradi@intel.com>:

Update LN and Softmax

--
62497479066d9d4a9017ad13ab2389c4795b37db by Sachin Muradi <sachin.muradi@intel.com>:

Address review comments

Merging this change closes #10211

FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/10211 from Intel-tensorflow:onednn-update-ln 62497479066d9d4a9017ad13ab2389c4795b37db
