[&](std::optional<ArrayRef<int64_t>> vals) -> DenseI64ArrayAttr {
      return vals.has_value()
                 ? DenseI64ArrayAttr::get(
                       ctx, InvertPermutationToVector(*vals, transpose_perm))
                 : nullptr;
    };
    DenseI64ArrayAttr new_window_dimensions =
        getInvertPermutedAttr(reduce_op.getWindowDimensions());
    DenseI64ArrayAttr new_window_strides =

  // `array_attr` and `permutation` must be equal. Returns a null attribute
  // if `array_attr` is null.
  DenseI64ArrayAttr PermuteI64ArrayAttr(
      PatternRewriter& rewriter,
      const std::optional<ArrayRef<int64_t>> array_attr,
      const ArrayRef<int64_t> permutation) const {
    if (!array_attr.has_value()) return DenseI64ArrayAttr(nullptr);

    return rewriter.getDenseI64ArrayAttr(

            op.getLoc(), op.getMean().getType(), alpha, op.getMean(),
            /*broadcast_dimensions=*/DenseI64ArrayAttr());
        auto beta_mul_batch_mean = rewriter.create<chlo::BroadcastMulOp>(
            op.getLoc(), batch_mean.getType(), beta, batch_mean,
            /*broadcast_dimensions=*/DenseI64ArrayAttr());
        batch_mean = rewriter.create<chlo::BroadcastAddOp>(