Skip to content

Commit 60ee9c6

Browse files
committed
docs(//cpp/api): Add documentation for input range, remove old struct
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 6a372af commit 60ee9c6

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

cpp/api/include/trtorch/trtorch.h

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,13 @@ namespace trtorch {
3838
*
3939
*/
4040
struct TRTORCH_API ExtraInfo {
41-
//struct TRTORCH_API InputRangesArray {
41+
/**
42+
* @brief A struct to hold an input range (used by TensorRT Optimization profile)
43+
*
44+
* This struct can either hold a single vector representing an input shape, signifying a
45+
* static input shape or a set of three input shapes representing the min, optiminal and max
46+
* input shapes allowed for the engine.
47+
*/
4248
struct TRTORCH_API InputRange {
4349
std::vector<int64_t> min;
4450
std::vector<int64_t> opt;

0 commit comments

Comments
 (0)