[oneDNN]: Fix high memory usage when running eager ops as a function.
Fixes https://github.com/tensorflow/tensorflow/issues/58676.

When running eager op as a function, `tf.data` ops are getting repeatedly cached leading to high memory usage as reported in the above issue. The fix is to recognize such ops and prevent them from getting cached again.