[oneDNN] Fix failing resnet50 benchmark tests with V2
This PR fixes the following 2 benchmark tests with v2
//tensorflow/python/eager/benchmarks/resnet50:resnet50_test_cpu
//tensorflow/python/eager/benchmarks/resnet50:hvp_test_cpu

The failure started after this commit: https://github.com/tensorflow/tensorflow/commit/b44d7918070a2a192645af7b0c96b0c4e399161e 