Pass symboltable to onnx shape inference

microsoft · Jun 16, 2022 · 9e5af64 · 9e5af64
1 parent 32d1e80
commit 9e5af64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/onnxruntime/core/graph/function_utils.cc b/onnxruntime/core/graph/function_utils.cc
@@ -267,9 +267,10 @@ std::unique_ptr<ONNX_NAMESPACE::OpSchema> CreateSchema(const std::string& functi
         std::unordered_map<std::string, const ONNX_NAMESPACE::FunctionProto*> map_copy(model_local_functions.begin(),
                                                                                        model_local_functions.end());
         std::unordered_map<std::string, TensorShapeProto> empty_map;
+        ONNX_NAMESPACE::shape_inference::SymbolTableImpl symbolTable;
         ONNX_NAMESPACE::shape_inference::InferShapeForFunctionNode(*onnx_func_proto, func_domain_to_version,
                                                                    schema_registry, ctx, options, map_copy,
-                                                                   nullptr, &empty_map);
+                                                                   &symbolTable, &empty_map);
       });
 
   op_schema->Finalize();

diff --git a/onnxruntime/core/graph/function_utils.h b/onnxruntime/core/graph/function_utils.h
@@ -2,6 +2,8 @@
 // Copyright (c) Microsoft Corporation. All rights reserved.
 // Licensed under the MIT License.
 
+#include <string>
+
 #include "core/common/common.h"
 #include "onnx/onnx_pb.h"
 #include "core/graph/graph.h"