saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-25 22:26:24 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added emotion ferplus support (#1752)

Browse source 
Signed-off-by: suryasidd <surya.siddharth.pemmaraju@intel.com>
This commit is contained in:
suryasidd 2019-09-03 15:01:22 -07:00 committed by jywu-msft
parent 94d9161166
commit 9523977cc2
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
onnxruntime/core/providers/openvino/openvino_execution_provider.cc
View file

@ -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";
}