tensorflow / community

Stores documents used by the TensorFlow developer community

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to enable oneddn in tensorflow for eltwise op

akote123 opened this issue · comments

@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