diff --git a/onnxruntime/core/providers/openvino/openvino_execution_provider.cc b/onnxruntime/core/providers/openvino/openvino_execution_provider.cc index f2a15562bb..f79578ce0e 100644 --- a/onnxruntime/core/providers/openvino/openvino_execution_provider.cc +++ b/onnxruntime/core/providers/openvino/openvino_execution_provider.cc @@ -180,7 +180,9 @@ bool IsOpSupported(std::string name) { "Unsqueeze", "ImageScaler", "LeakyRelu", - "GlobalMaxPool"}; + "GlobalMaxPool", + "Div", + "Sub"}; auto iter = supported_ops.find(name); return iter != supported_ops.end(); @@ -237,7 +239,10 @@ void CheckGraphSupported(const onnxruntime::GraphViewer& graph_viewer, std::stri //Zero dimension check for (size_t i = 0; i < node_inputs.size(); i++) { - if (node_inputs[i]->Shape() != nullptr) { + + auto name = node_inputs[i]->Name(); + auto it = initializers.find(name); + if(it == initializers.end() && node_inputs[i]->Shape() != nullptr){ if (node_inputs[i]->Shape()->dim_size() == 0) { throw "Node_input is zero dimension"; }