Use opsets in sample of function creation (#7792)

This commit is contained in:
Evgenya Stepyreva 2021-12-23 13:41:27 +03:00 committed by GitHub
parent 516272aeee
commit 41ace9d4e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -12,6 +12,7 @@
// clang-format off // clang-format off
#include "openvino/openvino.hpp" #include "openvino/openvino.hpp"
#include "openvino/opsets/opset1.hpp"
#include "openvino/opsets/opset8.hpp" #include "openvino/opsets/opset8.hpp"
#include "ngraph/util.hpp" #include "ngraph/util.hpp"
@ -108,7 +109,7 @@ std::shared_ptr<ov::Model> createNgraphFunction(const std::string& path_to_weigh
Shape padBeginShape{0, 0}; Shape padBeginShape{0, 0};
Shape padEndShape{0, 0}; Shape padEndShape{0, 0};
auto maxPoolingNodeFirst = std::make_shared<op::v1::MaxPool>(addNodeFirst->output(0), auto maxPoolingNodeFirst = std::make_shared<opset1::MaxPool>(addNodeFirst->output(0),
Strides{2, 2}, Strides{2, 2},
padBeginShape, padBeginShape,
padEndShape, padEndShape,
@ -138,7 +139,7 @@ std::shared_ptr<ov::Model> createNgraphFunction(const std::string& path_to_weigh
std::make_shared<opset8::Add>(convolutionNodeSecond->output(0), addSecondConstantNode->output(0)); std::make_shared<opset8::Add>(convolutionNodeSecond->output(0), addSecondConstantNode->output(0));
// -------MAXPOOL 2-------- // -------MAXPOOL 2--------
auto maxPoolingNodeSecond = std::make_shared<op::v1::MaxPool>(addNodeSecond->output(0), auto maxPoolingNodeSecond = std::make_shared<opset1::MaxPool>(addNodeSecond->output(0),
Strides{2, 2}, Strides{2, 2},
padBeginShape, padBeginShape,
padEndShape, padEndShape,
@ -152,7 +153,7 @@ std::shared_ptr<ov::Model> createNgraphFunction(const std::string& path_to_weigh
std::make_shared<opset8::Constant>(element::Type_t::i64, reshapeFirstShape, data + reshapeOffset); std::make_shared<opset8::Constant>(element::Type_t::i64, reshapeFirstShape, data + reshapeOffset);
auto reshapeFirstNode = auto reshapeFirstNode =
std::make_shared<op::v1::Reshape>(maxPoolingNodeSecond->output(0), reshapeFirstConstantNode->output(0), true); std::make_shared<opset8::Reshape>(maxPoolingNodeSecond->output(0), reshapeFirstConstantNode->output(0), true);
// -------MatMul 1--------- // -------MatMul 1---------
auto matMulFirstShape = Shape{500, 800}; auto matMulFirstShape = Shape{500, 800};
@ -179,7 +180,7 @@ std::shared_ptr<ov::Model> createNgraphFunction(const std::string& path_to_weigh
std::make_shared<opset8::Constant>(element::Type_t::i64, reshapeSecondShape, data + reshapeOffset); std::make_shared<opset8::Constant>(element::Type_t::i64, reshapeSecondShape, data + reshapeOffset);
auto reshapeSecondNode = auto reshapeSecondNode =
std::make_shared<op::v1::Reshape>(reluNode->output(0), reshapeSecondConstantNode->output(0), true); std::make_shared<opset8::Reshape>(reluNode->output(0), reshapeSecondConstantNode->output(0), true);
// -------MatMul 2--------- // -------MatMul 2---------
auto matMulSecondShape = Shape{10, 500}; auto matMulSecondShape = Shape{10, 500};