Skip to main content

coverage_check_shapes

Function coverage_check_shapes 

Source
pub(super) fn coverage_check_shapes(
    shapes: &[(usize, usize)],
) -> Vec<(usize, usize)>
Expand description

Selects the minimal set of shapes needed to verify that an ORT TRT EP cached profile covers all shapes in shapes.

ORT’s TRT EP stores engine profiles as per-dimension [min, max] ranges. Verifying complete coverage requires bounding all four dimension extremes independently. This function returns at most 4 representative shapes — one with min_batch, one with max_batch, one with min_seq, one with max_seq — deduplicated so the same shape is never run twice.

When the shard has only one unique extreme in a dimension (e.g. all shapes share the same batch size), the duplicates collapse and the returned set is smaller.