24 lines
1.1 KiB
C++
24 lines
1.1 KiB
C++
// Copyright (C) 2020 Intel Corporation
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
//
|
|
|
|
#include "ngraph_functions/builders.hpp"
|
|
|
|
namespace ngraph {
|
|
namespace builder {
|
|
|
|
std::shared_ptr<ngraph::Node> makeScatterElementsUpdate(const ngraph::Output<Node> &in,
|
|
const element::Type& indicesType,
|
|
const std::vector<size_t>& indicesShape,
|
|
const std::vector<size_t>& indices,
|
|
const ngraph::Output<Node> &update,
|
|
int axis) {
|
|
auto indicesNode = std::make_shared<ngraph::opset1::Constant>(indicesType, indicesShape, indices);
|
|
auto axis_node = std::make_shared<ngraph::opset1::Constant>(ngraph::element::Type_t::i32, ngraph::Shape{},
|
|
std::vector<int>{axis});
|
|
auto dtsNode = std::make_shared<ngraph::opset3::ScatterElementsUpdate>(in, indicesNode, update, axis_node);
|
|
return dtsNode;
|
|
}
|
|
|
|
} // namespace builder
|
|
} // namespace ngraph
|