[Intel oneDNN][Bug Fix] fix unit test failure with framework::function test
This PR fixes a unit test failure which was caused by the merge of this past PR

 https://github.com/tensorflow/tensorflow/pull/55411