gather_op.getDimensionNumbers().getCollapsedSliceDims().end());
  std::vector<int64_t> start_index_map_vec(
      gather_op.getDimensionNumbers().getStartIndexMap().begin(),
      gather_op.getDimensionNumbers().getStartIndexMap().end());

  auto offset_dims = builder_.CreateVector(offset_dims_vec);
  auto collapsed_slice_dims = builder_.CreateVector(collapsed_slice_dims_vec);

      return failure();
    }

    // Verify that start_index_map and collapsed_slice_dims contains the same
    // values.
    auto start_index_map = gather_op.getDimensionNumbers().getStartIndexMap();
    auto collapsed_slice_dims =
        gather_op.getDimensionNumbers().getCollapsedSliceDims();
    if (start_index_map.size() != collapsed_slice_dims.size()) {
      return rewriter.notifyMatchFailure(

    ::mlir::stablehlo::GatherDimensionNumbersAttr dim_numbers =
        op.getDimensionNumbers();
    ArrayRef<int64_t> offset_dims = dim_numbers.getOffsetDims();
    ArrayRef<int64_t> start_index_map = dim_numbers.getStartIndexMap();

    // Check for condition 1.
    if (start_index_map.empty() || start_index_map[0] != 0) {
      return failure();
    }
    for (int64_t i = 0; i < start_index_map.size() - 1; ++i) {