From 7c230dbc104c487db02f8fdb41f2ef841cfc086d Mon Sep 17 00:00:00 2001 From: liqun Fu Date: Fri, 8 Apr 2022 10:04:21 -0700 Subject: [PATCH] fix code error in function.cc (#11148) --- onnxruntime/core/graph/function.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/core/graph/function.cc b/onnxruntime/core/graph/function.cc index 3327ab0889..e7098c8e79 100644 --- a/onnxruntime/core/graph/function.cc +++ b/onnxruntime/core/graph/function.cc @@ -226,7 +226,7 @@ void IOTypeConstraintHelper(const ONNX_NAMESPACE::FunctionProto& onnx_func_proto // the requested types. auto& dest_types = type_constraint_map[type_str]; if (node_op_schema) { - const auto& types = node_op_schema->inputs().at(i).GetTypes(); + const auto& types = node_op_schema->outputs().at(i).GetTypes(); dest_types.reserve(dest_types.size() + types.size()); for (auto* data_type : types) { dest_types.emplace_back(*data_type);