[oneDNN v3.x]: Added support for batchnorm fwd and bwd for fp32 and bf16
Hi @penpornk Can you please review this PR ? Thank you! Hi @penpornk Can you please review this PR ? Thank you! @penpornk I have addressed your review comments. Please take a look. Thanks.
- This PR adds support for oneDNN v3.x in batchnorm fwd and bwd for fp32 and bf16.
- This PR passes all unit tests related to batchnorm fwd and bwd for fp32/bf16 when oneDNN v3.x is enabled.
- This PR does not add or affect Eigen ops and is specific only to oneDNN kernels.
- For additional comments related to oneDNN v3.x integration, please refer to [this](https://github.com/tensorflow/tensorflow/pull/60307#issue-1665423762) comment.