start_indices_vector.reserve(op.getStartIndices().size());
    Value clamp_min = rewriter.create<TF::ConstOp>(
        op.getLoc(),
        rewriter.getIntegerAttr(signed_start_indices_element_type, 0));
    for (uint64_t i = 0, e = op.getStartIndices().size(); i < e; ++i) {
      // Always put a cast there.
      auto start = op.getStartIndices()[i];

    ArrayRef<int64_t> idx_shape(rank);
    TensorType idx_type =
        operand_type.cloneWith(idx_shape, rewriter.getI32Type());

    ArrayRef<int64_t> start_idx_i64 = op.getStartIndices();
    ArrayRef<int64_t> limit_idx_i64 = op.getLimitIndices();

    SmallVector<int32_t> start_idx_i32 =
        CastI64ArrayToI32(start_idx_i64).value();

  if (input_shape == xla::Shape() || input_shape.is_unbounded_dynamic())
    return false;

  xla::Shape start_indices_shape =
      xla::TypeToShape(op.getStartIndices().getType());
  if (start_indices_shape == xla::Shape()) return false;

  xla::GatherDimensionNumbers gather_dim_numbers;
  if (!gather_dim_numbers.ParseFromString(op.getDimensionNumbers().str()))

  // Check if update replaces the whole tensor, meaning operand and update has
  // the same shape and all start indices are zero.
  DenseIntElementsAttr indices_attr;
  if (matchPattern(getStartIndices(), m_Constant(&indices_attr)) &&
      indices_attr.isSplat() && indices_attr.getSplatValue<int>() == 0 &&
      getOperand().getType().hasStaticShape() &&
      getUpdate().getType().hasStaticShape() &&