Conv2DCustomBackpropInputOp only supports NHWC.

tensorflow.python.framework.errors_impl.InvalidArgumentError: Conv2DCustomBackpropInputOp only supports NHWC.
[[node Conv2DBackpropInput (defined at /lib/python3.7/site-packages/tensorflow_core/python/framework/ops.py:1751) ]] [Op:__inference_distributed_function_20858]

andrew@1macbook:~/demo/models## run the example
export TENSORFLOW_MODELS=$(pwd)
export CIFAR_DATA=/tmp/cifar10_data
export PYTHONPATH=${
   TENSORFLOW_MODELS}:$PYTHONPATH
andrew@macbook:~/demo/models#
export PYTHONPATH=${
   PYTHONPATH}:${
   TENSORFLOW_MODELS}
export TF_CONFIG='{"cluster": { "chief": ["localhost:2222"], "worker": ["localhost:2223"]}, "task": {"type": "worker", "index": 0}}'                            
python3 resnet_cifar_main.py --data_dir=${
   CIFAR_DATA} --num_gpus=0 --ds=multi_worker_mirrored --train_epochs=1
/usr/local/Cellar/python/3.7.4_1/Frameworks/Python.framework/Versions/3.7/Resources/Python.app/Contents/MacOS/Python: can't open file 'resnet_cifar_main.py': [Errno 2] No such file or directory
andrew@2macbook:~/demo/models#cd ~/tensorflowonspark/TensorFlowOnSpark/examples/resnet
andrew@macbook:~/tensorflowonspark/TensorFlowOnSpark/examples/resnet#
export PYTHONPATH=${
   PYTHONPATH}:${
   TENSORFLOW_MODELS}
export TF_CONFIG='{"cluster": { "chief": ["localhost:2222"], "worker": ["localhost:2223"]}, "task": {"type": "worker", "index": 0}}'
python3 resnet_cifar_main.py --data_dir=${
   CIFAR_DATA} --num_gpus=0 --ds=multi_worker_mirrored --train_epochs=1
2020-05-25 14:10:22.263551: I tensorflow/core/platform/cpu_feature_guard.cc:142] Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA
2020-05-25 14:10:22.273714: I tensorflow/compiler/xla/service/service.cc:168] XLA service 0x7fb2ea468e10 executing computations on platform Host. Devices:
2020-05-25 14:10:22.273730: I tensorflow/compiler/xla/service/service.cc:175]   StreamExecutor device (0): Host, Default Version
2020-05-25 14:10:22.279316: I tensorflow/core/distributed_runtime/rpc/grpc_channel.cc:258] Initialize GrpcChannelCache for job chief -> {
   0 -> localhost:2222}
2020-05-25 14:10:22.279329: I tensorflow/core/distributed_runtime/rpc/grpc_channel.cc:258] Initialize GrpcChannelCache for job worker -> {
   0 -> localhost:2223}
2020-05-25 14:10:22.279803: I tensorflow/core/distributed_runtime/rpc/grpc_server_lib.cc:365] Started server with target: grpc://localhost:2223
INFO:tensorflow:Enabled multi-worker collective ops with available devices: ['/job:worker/replica:0/task:0/device:CPU:0', '/job:worker/replica:0/task:0/device:XLA_CPU:0']
I0525 14:10:22.280099 4604100032 collective_all_reduce_strategy.py:269] Enabled multi-worker collective ops with available devices: ['/job:worker/replica:0/task:0/device:CPU:0', '/job:worker/replica:0/task:0/device:XLA_CPU:0']
INFO:tensorflow:Multi-worker CollectiveAllReduceStrategy with cluster_spec = {
   'chief': ['localhost:2222'