tflite_support.metadata_writers.metadata_info.InputTextTensorMd

A container for the input text tensor metadata information.

Inherits From: TensorMd

View aliases

Main aliases

tflite_support.metadata_writers.audio_classifier.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.audio_classifier.metadata_writer.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.bert_nl_classifier.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.bert_nl_classifier.metadata_writer.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.image_classifier.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.image_classifier.metadata_writer.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.image_segmenter.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.image_segmenter.metadata_writer.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.nl_classifier.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.nl_classifier.metadata_writer.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.object_detector.metadata_info.InputTextTensorMd, tflite_support.metadata_writers.object_detector.metadata_writer.metadata_info.InputTextTensorMd

tflite_support.metadata_writers.metadata_info.InputTextTensorMd(
    name: Optional[str] = None,
    description: Optional[str] = None,
    tokenizer_md: Optional[tflite_support.metadata_writers.metadata_info.RegexTokenizerMd] = None
)

Args
name	name of the tensor.
description	description of what the tensor is.
tokenizer_md	information of the tokenizer in the input text tensor, if any. Only RegexTokenizer 1 is currenly supported. If the tokenizer is BertTokenizer 2 or SentencePieceTokenizer 3, refer to bert_nl_classifier.MetadataWriter.

Attributes
tokenizer_md	information of the tokenizer in the input text tensor, if any.

Methods

create_metadata

View source

create_metadata() -> tflite_support.metadata_schema_py_generated.TensorMetadataT

Creates the input text metadata based on the information.

Returns
A Flatbuffers Python object of the input text metadata.

Raises
ValueError	if the type of tokenizer_md is unsupported.