File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
plugin_execution_providers/tensorrt Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -1717,7 +1717,9 @@ OrtStatus* ORT_API_CALL TensorrtExecutionProvider::GetCapabilityImpl(OrtEp* this
17171717 int number_of_trt_nodes = 0 ;
17181718 for (const auto & group : supported_nodes_vector) {
17191719 if (!group.first .empty ()) {
1720- std::vector<const OrtNode*> supported_nodes (group.first .size ());
1720+ std::vector<const OrtNode*> supported_nodes;
1721+ supported_nodes.reserve (group.first .size ());
1722+
17211723 for (const auto & index : group.first ) {
17221724 const OrtNode* supported_node = nodes[index];
17231725
@@ -1782,7 +1784,7 @@ OrtStatus* ORT_API_CALL TensorrtExecutionProvider::CompileImpl(_In_ OrtEp* this_
17821784
17831785 // Gets number of node's outputs
17841786 size_t num_node_outputs = 0 ;
1785- RETURN_IF_ERROR (ort_api.Node_GetNumInputs (fused_node, &num_node_outputs));
1787+ RETURN_IF_ERROR (ort_api.Node_GetNumOutputs (fused_node, &num_node_outputs));
17861788
17871789 std::vector<const OrtValueInfo*> node_outputs (num_node_outputs);
17881790 RETURN_IF_ERROR (ort_api.Node_GetOutputs (fused_node, node_outputs.data (), node_outputs.size ()));
You can’t perform that action at this time.
0 commit comments