experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

NFC: Add a TODO to move HLO Relu legalizations to TF to TF lowering

Browse Source 
PiperOrigin-RevId: 296095163
Change-Id: Ic0b8d26c11c64e6584eef0da38b87e71d2dd03e8
This commit is contained in:
Smit Hinsu 2020-02-19 17:25:04 -08:00 committed by TensorFlower Gardener
parent 8b7a3db0b6
commit 7ad1eb110f
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/xla/transforms/legalize_tf_patterns.td
View File

@ -368,6 +368,9 @@ def : Pat<(TF_ConstOp:$res ElementsAttr:$value), (HLO_ConstOp $value),
// Relu op patterns.
//===----------------------------------------------------------------------===//
// TODO(hinsu): Make these patterns to TF to TF lowering. Relu6 lowering will
// require HLO canonicalization of min and max on a tensor to ClampOp.
// TODO(hinsu): Lower unsinged and quantized types after supporting
// them in GetScalarOfType.
def : Pat<(TF_ReluOp AnyRankedTensor:$input),