diff --git a/inference-engine/src/vpu/common/src/ngraph/utilities.cpp b/inference-engine/src/vpu/common/src/ngraph/utilities.cpp index 9bf6c21fb7d..1ef247aa2ee 100644 --- a/inference-engine/src/vpu/common/src/ngraph/utilities.cpp +++ b/inference-engine/src/vpu/common/src/ngraph/utilities.cpp @@ -49,13 +49,16 @@ ngraph::HostTensorVector evaluateOp(ngraph::Node* node, const ngraph::HostTensor std::vector evaluateTargetShape(const ngraph::Output& value) { static ngraph::Evaluator::op_handler_map handlers = { - {ngraph::opset3::ShapeOf::type_info, evaluateShapeOf}, - {ngraph::opset3::Constant::type_info, evaluateConstant}, - {ngraph::opset3::Gather::type_info, evaluateOp}, - {ngraph::opset3::Concat::type_info, evaluateOp}, - {ngraph::opset3::Reshape::type_info, evaluateOp}, - {ngraph::opset3::Multiply::type_info, evaluateOp}, - {ngraph::opset3::Squeeze::type_info, evaluateOp}, + {ngraph::opset3::ShapeOf::type_info, evaluateShapeOf}, + {ngraph::opset3::Constant::type_info, evaluateConstant}, + {ngraph::opset3::Gather::type_info, evaluateOp}, + {ngraph::opset3::Concat::type_info, evaluateOp}, + {ngraph::opset3::Reshape::type_info, evaluateOp}, + {ngraph::opset3::Multiply::type_info, evaluateOp}, + {ngraph::opset3::Squeeze::type_info, evaluateOp}, + {ngraph::opset5::Unsqueeze::type_info, evaluateOp}, + {ngraph::opset5::Equal::type_info, evaluateOp}, + {ngraph::opset5::Select::type_info, evaluateOp}, }; ngraph::Evaluator::value_map value_map; ngraph::Evaluator evaluator(handlers, value_map);