IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE][VPU]: Extend StaticShapeBroadcast target shape evaluator (#3561)

Browse Source 
Ticket - #-42237
Add Unsqueeze, Equal and Select operations to the StaticShapeBroadcast target shape evaluator, because they are presented in the target shape subgraph evaluator in one of the network, we are currently enabling.
This commit is contained in:
Andrew Bakalin 2020-12-11 12:30:15 +03:00 committed by GitHub
parent 578ea2fc3c
commit a0952798ba
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
inference-engine/src/vpu/common/src/ngraph/utilities.cpp
View File

@ -49,13 +49,16 @@ ngraph::HostTensorVector evaluateOp(ngraph::Node* node, const ngraph::HostTensor
std::vector<std::int64_t> evaluateTargetShape(const ngraph::Output<ngraph::Node>& value) { std::vector<std::int64_t> evaluateTargetShape(const ngraph::Output<ngraph::Node>& value) {
static ngraph::Evaluator<ngraph::HostTensorPtr>::op_handler_map handlers = { static ngraph::Evaluator<ngraph::HostTensorPtr>::op_handler_map handlers = {
{ngraph::opset3::ShapeOf::type_info, evaluateShapeOf}, {ngraph::opset3::ShapeOf::type_info, evaluateShapeOf},
{ngraph::opset3::Constant::type_info, evaluateConstant}, {ngraph::opset3::Constant::type_info, evaluateConstant},
{ngraph::opset3::Gather::type_info, evaluateOp}, {ngraph::opset3::Gather::type_info, evaluateOp},
{ngraph::opset3::Concat::type_info, evaluateOp}, {ngraph::opset3::Concat::type_info, evaluateOp},
{ngraph::opset3::Reshape::type_info, evaluateOp}, {ngraph::opset3::Reshape::type_info, evaluateOp},
{ngraph::opset3::Multiply::type_info, evaluateOp}, {ngraph::opset3::Multiply::type_info, evaluateOp},
{ngraph::opset3::Squeeze::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<ngraph::HostTensorPtr>::value_map value_map; ngraph::Evaluator<ngraph::HostTensorPtr>::value_map value_map;
ngraph::Evaluator<ngraph::HostTensorPtr> evaluator(handlers, value_map); ngraph::Evaluator<ngraph::HostTensorPtr> evaluator(handlers, value_map);