Skip to content

Commit 6cd5655

Browse files
authored
Update torch.mdx
1 parent d54582c commit 6cd5655

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

docs/backend-reference/torch.mdx

+3-3
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,7 @@ For onnx models, there are the following additional parameters:
128128

129129
| Parameter | Description | Note |
130130
|-------------------------|-----------------------------------------------------------------------------------|------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
131-
| min/max | Minimum/maximum input of the model | In form, it can be `1`, `1x3x224x224`, or `1,1` (for multi-input networks). Sorted in lexicographic order;<br />When `instance_num > 1`, it can be multiple configurations separated by `;`. |
131+
| min/max | Minimum/maximum input of the model | In form, it can be `1`, `1x3x224x224`, or `1,1` (for multi-input networks). <br />When `instance_num > 1`, it can be multiple configurations separated by `;`. |
132132
| precision | Model precision | [fp32,fp16,int8,best]. The default value for versions <=0.3.1b1 is fp16, and for versions >0.3.1b1 it is [fp16(SM>6.1), fp32(SM<=6.1)]; If the precision is not supported, it will automatically degrade to a supported precision. |
133133
| precision::fp32 | Set the precision of some layers to fp32 (overrides the precision setting) | Layer name(s) (can provide only part of the name), separated by commas. |
134134
| precision::fp16 | Set the precision of some layers to fp16 (overrides the precision setting) | Layer name(s) (can provide only part of the name), separated by commas. |
@@ -155,8 +155,8 @@ See [example](https://github.com/torchpipe/torchpipe/tree/main/examples/int8).
155155

156156
| | Description | Note |
157157
|--------------------------|-----------------------------------------------------------------------------------------------------------------------------------------------------------------|-------------------------------|
158-
| TASK_DATA_KEY | When the network has a single input and output, the type is at::Tensor/torch.Tensor. When the network has multiple inputs and outputs, the type is vector/List. | Sorted in lexicographic order |
159-
| TASK_RESULT_KEY (Output) | The output type is the same as the input type, and the postprocessor can customize the output. | Sorted in lexicographic order |
158+
| TASK_DATA_KEY | When the network has a single input and output, the type is at::Tensor/torch.Tensor. When the network has multiple inputs and outputs, the type is vector/List. | Sorted in lexicographic order for trt<=9 |
159+
| TASK_RESULT_KEY (Output) | The output type is the same as the input type, and the postprocessor can customize the output. | Sorted in lexicographic order for trt<=9 |
160160

161161

162162
### min()/max()

0 commit comments

Comments
 (0)