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:
parent
8c32f52323
commit
d0901a83e5
|
@ -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());
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue