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. // here for lowering to HLO.
TF::PopulateLoweringTFPatterns(context, &patterns); TF::PopulateLoweringTFPatterns(context, &patterns);
patterns.insert< patterns.insert<
ConvertArgMaxOp, ConvertBF16FloorDivOp, ConvertConv2D, ConvertAllOp, ConvertAnyOp, ConvertArgMaxOp, ConvertBF16FloorDivOp,
ConvertFusedBatchNormV3Op, ConvertEinsumOp, ConvertMaxPoolOp, ConvertConv2D, ConvertConv2DBackpropFilterOp,
ConvertRangeOp, ConvertSigmoidOp, ConvertSizeOp, ConvertMaxPoolOp, ConvertConv2DBackpropInputOp, ConvertEinsumOp, ConvertFusedBatchNormV3Op,
ConvertRangeOp, ConvertSigmoidOp, ConvertMaxOp, ConvertMaxPoolOp, ConvertMaxPoolGradOp, ConvertMeanOp,
ConvertOneHotOp, ConvertRangeOp, ConvertSigmoidOp, ConvertSizeOp,
ConvertSoftmaxOp<TF::LogSoftmaxOp, true>, ConvertSoftmaxOp<TF::LogSoftmaxOp, true>,
ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp, ConvertSoftmaxOp<TF::SoftmaxOp, false>, ConvertSplitOp, ConvertSplitVOp,
ConvertStridedSliceOp, ConvertStridedSliceGradOp, ConvertTopKV2Op, ConvertStridedSliceOp, ConvertStridedSliceGradOp, ConvertSumOp,
ConvertUnpackOp, ConvertMeanOp, ConvertSumOp, ConvertMaxOp, ConvertAllOp, ConvertTileOp, ConvertTopKV2Op, ConvertUnpackOp,
ConvertAnyOp, ConvertTileOp, ConvertMaxPoolGradOp, ConvertOneHotOp,
ConvertConv2DBackpropInputOp, ConvertConv2DBackpropFilterOp,
ConvertUnsortedSegmentMaxOp, ConvertUnsortedSegmentMinOp, ConvertUnsortedSegmentMaxOp, ConvertUnsortedSegmentMinOp,
ConvertUnsortedSegmentProdOp, ConvertUnsortedSegmentSumOp>( ConvertUnsortedSegmentProdOp, ConvertUnsortedSegmentSumOp>(
op->getContext()); op->getContext());