search

openxla / xla

A machine learning compiler for GPUs, CPUs, and ML accelerators
Apache License 2.0
2.64k stars 418 forks source link

[XLA:CPU][oneDNN] Add post-ops for oneDNN Convolutions #17855

Open akhilgoe opened 1 week ago

akhilgoe commented 1 week ago

This PR adds support for multiple post-ops for oneDNN Convolutions. New Post-ops added:

  1. Relu
  2. Tanh
  3. Gelu (Approx.)
  4. Gelu (Exact)
  5. Relu6
  6. Sigmoid
  7. Elu
  8. Elementwise scalar product

Additionally, this PR refactors the oneDNN convolution test file and adds parameterized tests for F32, BF16 and F16.