[CPU] Extensions shape inference issue fix (#18824)
This commit is contained in:
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;
|
||||
|
||||
+2
-1
@@ -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");
|
||||
}
|
||||
|
||||
|
||||
Reference in new issue
Block a user