docs: def piece_size(model_file=None, model_proto=None, name=None):
    """Returns the piece size (vocabulary size).
  
    Args:
      model_file: The sentencepiece model file path.
      model_proto: The sentencepiece model serialized proto.
                   Either `model_file` or `model_proto` must be set.
      name: The name argument that is passed to the op function.
    Returns:
      A scalar representing the vocabulary size.
    """
  
    return _gen_sentencepiece_processor_op.sentencepiece_get_piece_size(
        model_file=model_file, model_proto=model_proto, name=name)