Module: tflite_support.metadata_writers.writer_utils

View source on GitHub

Helper methods for writing metadata into TFLite models.

Functions

compute_flat_size(...): Computes the flat size (number of elements) of tensor shape.

get_input_tensor_names(...): Gets a list of the input tensor names.

get_input_tensor_shape(...): Gets the shape of the specified input tensor.

get_input_tensor_types(...): Gets a list of the input tensor types.

get_output_tensor_names(...): Gets a list of the output tensor names.

get_output_tensor_types(...): Gets a list of the output tensor types.

get_tokenizer_associated_files(...): Gets a list of associated files packed in the tokenzier_options.

load_file(...): Loads file from the file path.

save_file(...): Loads file from the file path.