colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

PiperOrigin-RevId: 362002943
This commit is contained in:
A. Unique TensorFlower 2021-03-10 02:20:45 -08:00 committed by TensorFlow MLIR Team
parent 218476128e
commit 507d9fb61d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Dialect/mhlo/transforms/transform_unranked_hlo.cc
View File

@ -58,7 +58,7 @@ namespace {
sep fn(SinhOp) sep fn(TanOp) sep fn(SinhOp) sep fn(TanOp)
// TODO(herhut): Generate these out of op definitions. // 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> template <typename OpTy>
inline void AddLegalOpOnRankedTensor(ConversionTarget *target) { inline void AddLegalOpOnRankedTensor(ConversionTarget *target) {