54c384db11
* Updated common migration pipeline * Fixed merge issue * Added new model and extended example * Fixed typo * Added v10-v11 comparison
49 lines
1.9 KiB
C++
49 lines
1.9 KiB
C++
// Copyright (C) 2018-2021 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
#include <ngraph/ngraph.hpp>
|
|
#include <ngraph/opsets/opset8.hpp>
|
|
|
|
int main() {
|
|
//! [ngraph:graph]
|
|
// _____________ _____________
|
|
// | Parameter | | Parameter |
|
|
// | data1 | | data2 |
|
|
// |___________| |___________|
|
|
// | |
|
|
// data1_t | | data2_t
|
|
// \ /
|
|
// \ /
|
|
// \ /
|
|
// ____\____/____
|
|
// | Concat |
|
|
// | concat |
|
|
// |____________|
|
|
// |
|
|
// | concat_t
|
|
// |
|
|
// _______|_______
|
|
// | Result |
|
|
// | result |
|
|
// |_____________|
|
|
auto data1 = std::make_shared<ngraph::opset8::Parameter>(ngraph::element::i64, ngraph::Shape{1, 3, 2, 2});
|
|
data1->set_friendly_name("data1"); // operation name
|
|
data1->output(0).set_names({"data1_t"}); // tensor names
|
|
auto data2 = std::make_shared<ngraph::opset8::Parameter>(ngraph::element::i64, ngraph::Shape{1, 2, 2, 2});
|
|
data2->set_friendly_name("data2"); // operation name
|
|
data2->output(0).set_names({"data2_t"}); // tensor names
|
|
|
|
auto concat = std::make_shared<ngraph::opset8::Concat>(ngraph::OutputVector{data1, data2}, 1);
|
|
concat->set_friendly_name("concat"); // operation name
|
|
concat->output(0).set_names({"concat_t"}); // tensor name
|
|
|
|
auto result = std::make_shared<ngraph::opset8::Result>(concat);
|
|
result->set_friendly_name("result"); // operation name
|
|
|
|
auto f = std::make_shared<ngraph::Function>(ngraph::ResultVector{result},
|
|
ngraph::ParameterVector{data1, data2},
|
|
"function_name");
|
|
//! [ngraph:graph]
|
|
return 0;
|
|
}
|