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

[LPT] Revert changes in fold_reshape (#17068)

Browse Source
This commit is contained in:
Vladislav Golubev
2023-04-20 12:43:59 +02:00
committed by GitHub
parent e53fc86988
commit f100c36ac9
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/common/low_precision_transformations/include/low_precision/network_helper.hpp
View File

@@ -321,7 +321,9 @@ std::shared_ptr<Node> fold_reshape(Args&&... args) {
const auto data_const = ov::as_type_ptr<opset1::Constant>(node->get_input_node_shared_ptr(0));
const auto target_shape = ov::as_type_ptr<opset1::Constant>(node->get_input_node_shared_ptr(1));
if (data_const && target_shape) {
return std::make_shared<opset1::Constant>(*data_const, node->get_output_shape(0));
return std::make_shared<opset1::Constant>(node->get_input_element_type(0),
node->get_output_shape(0),
data_const->get_data_ptr());
}
return fold<opset1::Reshape>(std::forward<Args>(args)...);
}