message: Only define functor for Polygamma on GPUs if MLIR generated kernels are
  disabled.
diff: @@ -21,8 +21,8 @@ namespace tensorflow {
  namespace functor {
  #if !defined(MLIR_GENERATED_GPU_KERNELS_ENABLED)
  DEFINE_BINARY2(zeta, float, double);
  -#endif
  DEFINE_BINARY2(polygamma, float, double);
  +#endif
  } // namespace functor
  } // namespace tensorflow
  