[MLIR][KernelGen] Add `tf.Polygamma` kernel

PiperOrigin-RevId: 362002943
2021-03-10 02:20:45 -08:00 · 2021-03-10 02:20:45 -08:00 · 507d9fb61d
parent 218476128e
commit 507d9fb61d
1 changed files with 1 additions and 1 deletions
--- 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) {