We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6a372af commit 60ee9c6Copy full SHA for 60ee9c6
cpp/api/include/trtorch/trtorch.h
@@ -38,7 +38,13 @@ namespace trtorch {
38
*
39
*/
40
struct TRTORCH_API ExtraInfo {
41
- //struct TRTORCH_API InputRangesArray {
+ /**
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
+ */
48
struct TRTORCH_API InputRange {
49
std::vector<int64_t> min;
50
std::vector<int64_t> opt;
0 commit comments