}

    // Erases variable arguments from `batch_func_op`. `erase_indices` contains
    // the indices of the arguments to erase.
    void eraseArguments(const BitVector& erase_indices);

    // Gets the argument operands to the called function.
    operand_range getArgOperands() {
      // Merge `in_tensors` and `captured_tensors`. `operandSegmentSizes`