colin
/
mlir-hlo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Enable fallback legalization for MaxPoolGradGrad and MaxPool3DGradGrad ops 

Requires,
* Override for ReducePrecision in HloMlirBuilder
* Sinking of constants for ReduceWindow op

PiperOrigin-RevId: 342330848
This commit is contained in:
Smit Hinsu 2020-11-13 13:48:17 -08:00 committed by TensorFlow MLIR Team
parent 8c32f52323
commit d0901a83e5
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Dialect/mhlo/transforms/sink_constants_to_control_flow.cc
View File

@ -50,6 +50,8 @@ class SinkConstantsToControlFlowPass
} else if (auto if_op = llvm::dyn_cast<IfOp>(op)) { } else if (auto if_op = llvm::dyn_cast<IfOp>(op)) {
SinkToRegion(&if_op.true_branch()); SinkToRegion(&if_op.true_branch());
SinkToRegion(&if_op.false_branch()); SinkToRegion(&if_op.false_branch());
} else if (auto reduce_window_op = llvm::dyn_cast<ReduceWindowOp>(op)) {
SinkToRegion(&reduce_window_op.body());
} else if (auto sort_op = llvm::dyn_cast<SortOp>(op)) { } else if (auto sort_op = llvm::dyn_cast<SortOp>(op)) {
SinkToRegion(&sort_op.comparator()); SinkToRegion(&sort_op.comparator());
} }