message: [ROCm] Re-enabling ROCm support for the average pooling op
  ROCm support for the average pooling op was disabled in PR #, because it was leading to a compiler error in the `--config=rocm` build. That issue has now been fixed and hence filign this PR to re-enable ROCm support for the average pooling op.
diff: @@ -36,10 +36,10 @@ limitations under the License.
  #include "tensorflow/core/util/padding.h"
  #include "tensorflow/core/util/tensor_format.h"
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  #include "tensorflow/core/kernels/maxpooling_op_gpu.h"
  #include "tensorflow/core/kernels/pooling_ops_common_gpu.h"
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  namespace tensorflow {
  @@ -112,7 +112,7 @@ REGISTER_KERNEL_BUILDER(
  Name("AvgPool").Device(DEVICE_CPU).TypeConstraint<Eigen::half>("T"),
  AvgPoolingOp<CPUDevice, Eigen::half>);
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  template <typename T>
  class AvgPoolingOp<GPUDevice, T> : public UnaryOp<T> {
  public:
  @@ -205,7 +205,7 @@ REGISTER_KERNEL_BUILDER(
  REGISTER_KERNEL_BUILDER(
  Name("AvgPool").Device(DEVICE_GPU).TypeConstraint<double>("T"),
  AvgPoolingOp<GPUDevice, double>);
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  // The operation to compute AvgPool gradients.
  // It takes two inputs:
  @@ -368,7 +368,7 @@ TF_CALL_float(REGISTER_CPU_KERNEL);
  TF_CALL_double(REGISTER_CPU_KERNEL);
  TF_CALL_half(REGISTER_CPU_KERNEL);
  -#if GOOGLE_CUDA
  +#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  // A CUDNN based AvgPoolingGrad implementation. It includes the padding as the
  // candidates for the pooling operation.
  @@ -577,6 +577,6 @@ REGISTER_KERNEL_BUILDER(Name("AvgPoolGrad")
  .HostMemory("orig_input_shape"),
  AvgPoolingGradOpCustomGPUKernel<Eigen::half>);
  -#endif // GOOGLE_CUDA
  +#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
  } // namespace tensorflow
  