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

Remove redundant and sort TF -> HLO legalization patterns.

Browse Source 
ConvertMaxPoolOp, ConvertRangeOp, and ConvertSigmoidOp were added to the patterns list twice.

PiperOrigin-RevId: 286300831
Change-Id: I9106e8fa3c275b9ed78f5176715d0a803463f1e4
This commit is contained in:
Andy Ly 2019-12-18 17:44:08 -08:00 committed by TensorFlower Gardener
parent 3e2c108708
commit c56549d2a6
1 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tensorflow/compiler/mlir/xla/transforms/legalize_tf.cc
View File

@ -2574,16 +2574,15 @@ LogicalResult legalizeTF(Operation *op, bool allow_partial_conversion) {
// here for lowering to HLO.
TF::PopulateLoweringTFPatterns(context, &patterns);
patterns.insert<
ConvertArgMaxOp, ConvertBF16FloorDivOp, ConvertConv2D,
ConvertFusedBatchNormV3Op, ConvertEinsumOp, ConvertMaxPoolOp,
ConvertRangeOp, ConvertSigmoidOp, ConvertSizeOp, ConvertMaxPoolOp,
ConvertRangeOp, ConvertSigmoidOp,
ConvertAllOp, ConvertAnyOp, ConvertArgMaxOp, ConvertBF16FloorDivOp,
ConvertConv2D, ConvertConv2DBackpropFilterOp,
ConvertConv2DBackpropInputOp, ConvertEinsumOp, ConvertFusedBatchNormV3Op,
ConvertMaxOp, ConvertMaxPoolOp, ConvertMaxPoolGradOp, ConvertMeanOp,
ConvertOneHotOp, ConvertRangeOp, ConvertSigmoidOp, ConvertSizeOp,
ConvertSoftmaxOp<TF::LogSoftmaxOp, true>,
ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp,
ConvertStridedSliceOp, ConvertStridedSliceGradOp, ConvertTopKV2Op,
ConvertUnpackOp, ConvertMeanOp, ConvertSumOp, ConvertMaxOp, ConvertAllOp,
ConvertAnyOp, ConvertTileOp, ConvertMaxPoolGradOp, ConvertOneHotOp,
ConvertConv2DBackpropInputOp, ConvertConv2DBackpropFilterOp,
ConvertStridedSliceOp, ConvertStridedSliceGradOp, ConvertSumOp,
ConvertTileOp, ConvertTopKV2Op, ConvertUnpackOp,
ConvertUnsortedSegmentMaxOp, ConvertUnsortedSegmentMinOp,
ConvertUnsortedSegmentProdOp, ConvertUnsortedSegmentSumOp>(
op->getContext());