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

Correct non ConversionRewriter transformation in LegalizeToLinalg

Browse Source 
PiperOrigin-RevId: 347622657
Change-Id: Id171aa6454f72cc0a57eee14953b1354d62c162f
This commit is contained in:
Tres Popp 2020-12-15 08:56:01 -08:00 committed by TensorFlower Gardener
parent 282c83de94
commit ad82a249b4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/legalize_to_linalg.cc
View File

@ -813,7 +813,8 @@ class ReduceConverter : public OpConversionPattern<lmhlo::ReduceOp> {
loc, /*resultTensorTypes=*/ArrayRef<Type>{}, loc, /*resultTensorTypes=*/ArrayRef<Type>{},
/*inputs=*/adaptor.operands(), /*outputBuffers=*/adaptor.out(), /*inputs=*/adaptor.operands(), /*outputBuffers=*/adaptor.out(),
/*initTensors=*/ValueRange{}, maps, types); /*initTensors=*/ValueRange{}, maps, types);
linalg_op.region().takeBody(reduce_op.body()); rewriter.inlineRegionBefore(reduce_op.body(), linalg_op.region(),
linalg_op.region().end());
{ {
OpBuilder::InsertionGuard region_guard(rewriter); OpBuilder::InsertionGuard region_guard(rewriter);
Block* block = linalg_op.getBody(); Block* block = linalg_op.getBody();