tensorflow / community

Stores documents used by the TensorFlow developer community
Apache License 2.0
1.26k stars 576 forks source link

How to enable oneddn in tensorflow for eltwise op #451

Open akote123 opened 1 year ago

akote123 commented 1 year ago

@penpornk , I wanted to understand the latest tensorflow supports eltwise onednn flow and also is there any document which I can refer for the onednn ops which are supported in latest tensorflow. Actually I wanted to understand the onednn flow for elewise op, so I tried profile the code for

import tensorflow as tf

x= tf.constant([-10, -5, 0.0, 5, 10], dtype = tf.float32)
y=tf.nn.relu(x)
print(y)

But this will not create onednn flow.

How we can generate onednn flow for eltwise ops through tensorflow framework?

Thanks