return failure();
      }

      // Transform updates so that update window dims are the trailing
      // dimensions in the update tensor.
      auto update_window_dims = scatter_dimension_numbers.getUpdateWindowDims();
      if (failed(CanonicalizeScatterUpdates(scatter_op, update_window_dims,
                                            indices, indices_type, new_updates,

  mlir::stablehlo::ScatterDimensionNumbersAttr scatter_dimension_numbers =
      scatter_op.getScatterDimensionNumbers();
  llvm::ArrayRef<int64_t> update_window_dims_mlir =
      scatter_dimension_numbers.getUpdateWindowDims();
  llvm::ArrayRef<int64_t> inserted_window_dims_mlir =
      scatter_dimension_numbers.getInsertedWindowDims();
  llvm::ArrayRef<int64_t> scatter_dims_to_operand_dims_mlir =