Function torch_tensorrt::executorch_backend::dims_fit#

Function Documentation#

inline bool torch_tensorrt::executorch_backend::dims_fit(const nvinfer1::Dims &dims, const InputProfileBounds &bounds)#

Whether one input shape falls inside one profile’s envelope.

Parameters:
  • dims – The runtime shape of the input.

  • bounds – The profile’s [min, max] envelope for that input.

Returns:

true when the ranks match and every extent is in range.