IntenseWebs/openvino
3
0 Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Gather->ShapeOf optimization fix (#6006)

Browse Source
This commit is contained in:
Evgenya Stepyreva authored and GitHub committed 2021-06-03 12:55:48 +03:00
1 parent b2a6cef0dc
commit 5494280dfa
1 file changed
+2 -2
Download Patch File Download Diff File Expand all files Collapse all files
inference-engine/src/transformations/src/transformations/common_optimizations/algebraic_simplification.cpp
+2 -2
View File
@@ -93,7 +93,7 @@ static bool simplify_gather_shapeof(shared_ptr<Node> node) {
auto zero_axis = opset3::Constant::create<int64_t>(element::i64, Shape{}, {0});
NodeVector new_ops;
auto new_shapeof = make_shared<opset3::ShapeOf>(gather->input_value(0));
auto new_shapeof = make_shared<opset3::ShapeOf>(gather->input_value(0), node->get_output_element_type(0));
new_ops.push_back(new_shapeof);
std::shared_ptr<Node> replace_op;
if (indices_rank.get_length() == 0) {
@@ -113,7 +113,7 @@ static bool simplify_gather_shapeof(shared_ptr<Node> node) {
new_ops.push_back(gather);
concat_inputs.push_back(gather);
}
auto shapeof_indices = make_shared<opset3::ShapeOf>(gather->input_value(1));
auto shapeof_indices = make_shared<opset3::ShapeOf>(gather->input_value(1), node->get_output_element_type(0));
new_ops.push_back(shapeof_indices);
concat_inputs.push_back(shapeof_indices);