builder.setInsertionPoint(init_op);
        builder.create<TF::InitializeTableFromTextFileV2Op>(
            init_op.getLoc(), init_op.getTableHandle(), const_op.getResult(),
            init_op.getKeyIndex(), init_op.getValueIndex(),
            init_op.getVocabSize(), init_op.getDelimiter());
        init_op.erase();
      }
    }
    func.eraseArguments(args_to_erase);
  }
}

}  // namespace

    }

    @Nullable
    private Object getCandidateValue(int c, int a) {
        return candidateValues[getValueIndex(c, a)];
    }

    private void setCandidateValue(int c, int a, @Nullable Object value) {
        candidateValues[getValueIndex(c, a)] = value;
    }

    private int getValueIndex(int c, int a) {
        return c * requestedAttributes.size() + a;
    }

    private enum MatchResult {

    //
    // In the above case, the delimiter will be not used since the key is just a
    // whole line and value is a line number.
    if (op.getKeyIndex() != kTextFileIndex_WholeLine ||
        op.getValueIndex() != kTextFileIndex_LineNumber) {
      return failure();
    }

    // Try to find filename from constant op.
    DenseStringElementsAttr filename_attr;
    if (!matchPattern(op.getFilename().getDefiningOp(),