trtexec: Use getNbOutputs when computing outputs

janbernloehr · nvpohanh · commit 6e5bebd95a38 · 2023-06-08T09:21:30.000+08:00
Signed-off-by: Jan Bernlöhr &lt;jan@bernloehrs.de&gt;
diff --git a/samples/common/sampleEngines.cpp b/samples/common/sampleEngines.cpp
@@ -206,7 +206,7 @@ void setTensorScalesFromCalibration(nvinfer1::INetworkDefinition& network, std::
             input->setDynamicRange(-127 * calibScale, 127 * calibScale);
         }
     }
-    bool const broadcastOutputFormats = broadcastIOFormats(outputFormats, network.getNbInputs());
+    bool const broadcastOutputFormats = broadcastIOFormats(outputFormats, network.getNbOutputs());
     for (int32_t i = 0, n = network.getNbOutputs(); i < n; ++i)
     {
         int32_t formatIdx = broadcastOutputFormats ? 0 : i;

Original file line number	Diff line number	Diff line change
`@@ -206,7 +206,7 @@ void setTensorScalesFromCalibration(nvinfer1::INetworkDefinition& network, std::`
`206`	`206`	`input->setDynamicRange(-127 * calibScale, 127 * calibScale);`
`207`	`207`	`}`
`208`	`208`	`}`
`209`		`- bool const broadcastOutputFormats = broadcastIOFormats(outputFormats, network.getNbInputs());`
	`209`	`+ bool const broadcastOutputFormats = broadcastIOFormats(outputFormats, network.getNbOutputs());`
`210`	`210`	`for (int32_t i = 0, n = network.getNbOutputs(); i < n; ++i)`
`211`	`211`	`{`
`212`	`212`	`int32_t formatIdx = broadcastOutputFormats ? 0 : i;`