[CPU] Extensions shape inference issue fix (#18824)

This commit is contained in:
Vladislav Golubev authored and GitHub committed 2023-07-28 15:02:38 +04:00
1 parent 9177809bd7
commit 5e846a0433
2 files changed
+3 -2

No files matched your search

@@ -33,7 +33,7 @@ public:
return m_shape_infer->get_pads_end();
}
port_mask_t get_port_mask() const override {
return m_port_mask & m_shape_infer->get_port_mask();
return m_port_mask;
}
private:
std::shared_ptr<IStaticShapeInfer> m_shape_infer;
@@ -89,9 +89,10 @@ protected:
auto inputParams = ngraph::builder::makeDynamicParams(ngPrc, inputDynamicShapes);
auto paramOuts = ngraph::helpers::convert2OutputVector(ngraph::helpers::castOps2Nodes<ngraph::op::Parameter>(inputParams));
auto customOp = std::make_shared<CustomOp>(paramOuts);
auto shapeOf = std::make_shared<ov::opset10::ShapeOf>(customOp->output(1));
ngraph::ResultVector results{std::make_shared<ngraph::opset3::Result>(customOp->output(0)),
std::make_shared<ngraph::opset3::Result>(customOp->output(1))};
std::make_shared<ngraph::opset3::Result>(shapeOf)};
function = std::make_shared<ngraph::Function>(results, inputParams, "customOpTest");
}