DimensionInputInfo

class nvtripy.DimensionInputInfo(value_bounds: Tuple[int | DimensionSize, int | DimensionSize, int | DimensionSize])[source]

Bases: object

Captures information about a dimension size input to a compiled function.

Parameters:

value_bounds (Tuple[int | DimensionSize, int | DimensionSize, int | DimensionSize]) – The value bound of the dimension size input, consisting of minimum, optimum, and maximum values.

Example: Dimension Size Input
1# The dimension size will support values in the range [1, 3],
2# optimizing for a size of 2.
3dim_inp = tp.DimensionInputInfo((1, 2, 3))
Local Variables
>>> dim_inp
DimensionInputInfo(min=(1,), opt=(2,), max=(3,))