tflite_support.metadata_writers.writer_utils.compute_flat_size

Stay organized with collections Save and categorize content based on your preferences.

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

tflite_support.metadata_writers.writer_utils.compute_flat_size(
    tensor_shape: Optional['array.array[int]']
) -> int

tensor_shape an array of the tensor shape values.

The flat size of the tensor shape. Return 0 if tensor_shape is None.