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.