}

  LogicalResult MatchMaxPoolReduceWindowOp(
      mlir::stablehlo::ReduceWindowOp op) const {
    // TODO: b/321099943 - Support explicit padding.
    if (HasPadding(op)) return failure();

    // Check that the reduce-window body is a max operation.
    return success(IsMaxFunction(op.getBody().front()));
  }