[INTEL oneDNN][Bug Fix]Use correct data type for dim size
All 3 CI failures are unrelated.
* [Py+CPP Ubuntu CPU](https://source.cloud.google.com/results/invocations/3fdb42ac-da26-4a19-a735-e8f1aa45c9e8/log): Tests timed out.
* [Py+CPP Ubuntu GPU](https://source.cloud.google.com/results/invocations/5280cdca-cf13-4318-a10c-2eb1b0beabaa/log): Tests timed out.
* [AMD ROCm](http://ml-ci.amd.com:21096/blue/organizations/jenkins/tensorflow%2Fgithub-prs-upstream-master%2FAMD-ROCm-Community-CI-Build/detail/PR-61346/2/pipeline): Error unrelated to this PR.
```
Analyzing: target //tensorflow/tools/pip_package:build_pip_package (648 packages loaded, 30206 targets configured)

ERROR: /workspace/tensorflow/compiler/xla/service/gpu/BUILD:853:11: in deps attribute of cc_library rule //tensorflow/compiler/xla/service/gpu:gpu_executable: Label '//tensorflow/tsl/platform:random' is duplicated

ERROR: /workspace/tensorflow/compiler/xla/service/gpu/BUILD:853:11: Analysis of target '//tensorflow/compiler/xla/service/gpu:gpu_executable' failed
```
This is a follow-up PR of https://github.com/tensorflow/tensorflow/pull/60568

This PR makes sure that all occurrences of dim size usage is tied with int64_t data type (versus int)
within all oneDNN kernel op implementation.