From 507d9fb61d273dda2f62fb8b74b963f579b62f1b Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Wed, 10 Mar 2021 02:20:45 -0800
Subject: [PATCH] [MLIR][KernelGen] Add `tf.Polygamma` kernel

PiperOrigin-RevId: 362002943
---
 lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
index 9325d7d..ded4ee3 100644
--- a/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
+++ b/lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
@@ -58,7 +58,7 @@ namespace {
               sep fn(SinhOp) sep fn(TanOp)
 
 // TODO(herhut): Generate these out of op definitions.
-#define MAP_CHLO_OPERATION_CWISE_BINARY(fn, sep) fn(ZetaOp)
+#define MAP_CHLO_OPERATION_CWISE_BINARY(fn, sep) fn(PolygammaOp) sep fn(ZetaOp)
 
 template <typename OpTy>
 inline void AddLegalOpOnRankedTensor(ConversionTarget *target) {