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

NFC: Remove redundant getOperation calls

Browse Source 
PiperOrigin-RevId: 268621083
This commit is contained in:
Smit Hinsu 2019-09-11 23:43:00 -07:00 committed by TensorFlower Gardener
parent eb539185b9
commit 73bb698461
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -243,7 +243,7 @@ class ConvertMaxPoolOp : public OpRewritePattern<TF::MaxPoolOp> {
/*paddings=*/DenseIntElementsAttr()); /*paddings=*/DenseIntElementsAttr());
BuildReduceBody<xla_hlo::MaxOp>(element_type, &reduce.body(), &rewriter); BuildReduceBody<xla_hlo::MaxOp>(element_type, &reduce.body(), &rewriter);
rewriter.replaceOp(op.getOperation(), reduce.getResult(0)); rewriter.replaceOp(op, reduce.getResult(0));
return matchSuccess(); return matchSuccess();
} }
}; };
@ -338,8 +338,8 @@ class ConvertSoftmaxOp : public OpRewritePattern<TF::SoftmaxOp> {
// Convert the summation result back to the original element type and divide // Convert the summation result back to the original element type and divide
// exponentials by the summations. // exponentials by the summations.
sum = rewriter.create<xla_hlo::ConvertOp>(loc, reduce_out_type, sum); sum = rewriter.create<xla_hlo::ConvertOp>(loc, reduce_out_type, sum);
rewriter.replaceOpWithNewOp<xla_hlo::DivOp>(op.getOperation(), op.getType(), rewriter.replaceOpWithNewOp<xla_hlo::DivOp>(op, op.getType(), exp, sum,
exp, sum, batch_dims); batch_dims);
return matchSuccess(); return matchSuccess();
} }
}; };