Regression: //bazel_pip/tensorflow/python/kernel_tests/nn_ops:pooling_ops_3d_test fails
cc: @TensorFlow-MKL  @vufg  Thanks for the ping, will roll it back shortly Fixed by https://github.com/tensorflow/tensorflow/commit/f0d77e35b28693d37f14f48ec8ed3c0133b8a09b Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60500">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60500">No</a>

<details><summary>Click to expand!</summary> 
 
 ### Issue Type

Bug

### Have you reproduced the bug with TF nightly?

Yes

### Source

source

### Tensorflow Version

git HEAD

### Custom Code

No

### OS Platform and Distribution

Ubuntu 20.04

### Mobile device

n/a

### Python version

3.10

### Bazel version

5.3.0

### GCC/Compiler version

10.2.1

### CUDA/cuDNN version

n/a

### GPU model and memory

n/a

### Current Behaviour?

https://github.com/tensorflow/tensorflow/commit/f9becdcb81053e80b9c9034f3d4c31984f161dd1 introduced unit test failure when TF_ENABLE_ONEDNN_OPTS=1

### Standalone code to reproduce the issue

```shell
bazel test --build_tests_only --config=mkl_aarch64_threadpool --copt=-flax-vector-conversions --test_env=TF_ENABLE_ONEDNN_OPTS=1 --test_env=TF2_BEHAVIOR=1 --define=no_tensorflow_py_deps=true --test_lang_filters=py --flaky_test_attempts=3 --test_size_filters=small,medium --test_output=errors --verbose_failures=true --test_keep_going --notest_verbose_timeout_warnings --local_test_jobs=64 --test_tag_filters=-nopip,-no_pip,-oss_serial,-no_oss,-oss_excluded,-v1only,-benchmark-test,-no_aarch64,-no_oss_py38,-no_oss_py39,-no_oss_py310 -k -- //bazel_pip/tensorflow/... -//bazel_pip/tensorflow/compiler/tf2tensorrt/... -//bazel_pip/tensorflow/compiler/xrt/... -//bazel_pip/tensorflow/core/tpu/... -//bazel_pip/tensorflow/go/... -//bazel_pip/tensorflow/java/... -//bazel_pip/tensorflow/python/integration_testing/... -//bazel_pip/tensorflow/tools/toolchains/... -//bazel_pip/tensorflow/lite/... -//bazel_pip/tensorflow/python/kernel_tests/nn_ops:atrous_conv2d_test
```


### Relevant log output

```shell
======================================================================
FAIL: testAvgPool3dGradInvalidKsize (__main__.PoolingTest)
PoolingTest.testAvgPool3dGradInvalidKsize
----------------------------------------------------------------------
tensorflow.python.framework.errors_impl.InvalidArgumentError: {{function_node __wrapped____MklNativeAvgPool3DGrad_device_/job:localhost/replica:0/task:0/device:CPU:0}} Sliding window ksize for dimension 1 was zero. [Op:AvgPool3DGrad]

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "/tmpfs/bazel_output/_bazel_ubuntu/eab0d61a99b6696edb3d2aff87b585e8/execroot/org_tensorflow/bazel-out/aarch64-opt/bin/bazel_pip/tensorflow/python/kernel_tests/nn_ops/pooling_ops_3d_test_cpu.runfiles/org_tensorflow/bazel_pip/tensorflow/python/kernel_tests/nn_ops/pooling_ops_3d_test.py", line 181, in testAvgPool3dGradInvalidKsize
    with self.assertRaisesRegex(
AssertionError: "ksize must be positive, got: *" does not match "{{function_node __wrapped____MklNativeAvgPool3DGrad_device_/job:localhost/replica:0/task:0/device:CPU:0}} Sliding window ksize for dimension 1 was zero. [Op:AvgPool3DGrad]"

----------------------------------------------------------------------
Ran 33 tests in 22.409s

FAILED (failures=1, skipped=1)
================================================================================
```
</details>