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

View File

@ -50,6 +50,8 @@ class SinkConstantsToControlFlowPass
} else if (auto if_op = llvm::dyn_cast<IfOp>(op)) {
SinkToRegion(&if_op.true_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)) {
SinkToRegion(&sort_op.comparator());
}