fix code error in function.cc (#11148)

This commit is contained in:
liqun Fu 2022-04-08 10:04:21 -07:00 committed by Chi Lo
parent 901a73d7f0
commit 7c230dbc10

View file

@ -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);