Skip to content

Commit 5b62d13

Browse files
authored
Avoid skipping subgraphs for non tf2onnx subgraphs
1 parent 2b989a2 commit 5b62d13

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

onnxruntime/core/providers/openvino/ov_versions/capability.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,11 @@ GetCapability::GetCapability(const GraphViewer& graph_viewer_param, std::string
4040

4141
std::vector<std::unique_ptr<ComputeCapability>> GetCapability::Execute() {
4242
std::vector<std::unique_ptr<ComputeCapability>> result;
43+
44+
// Check if it is a subgraph
45+
if (graph_viewer_.IsSubgraph() && graph_viewer_.Name() == "tf2onnx") {
46+
return result;
47+
}
4348

4449
// This is a list of initializers that nGraph considers as constants. Example weights, reshape shape etc.
4550
std::unordered_set<std::string> ng_required_initializers;

0 commit comments

Comments
 (0)