trtexec: Use getNbOutputs when computing outputs

Signed-off-by: Jan Bernlöhr <[email protected]>
xhdhr10000 · Jun 8, 2023 · 6e5bebd · 6e5bebd
1 parent f82a322
commit 6e5bebd
Showing 1 changed file with 1 addition and 1 deletion.
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;