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

[CPU] Fixed failed tests (#6742)

Browse Source
This commit is contained in:
Egor Shulman 2021-09-24 09:37:54 +03:00 committed by GitHub
parent ae3e3af521
commit 95342de8c8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/mkldnn_plugin/mkldnn_plugin.cpp
View File

@ -656,11 +656,11 @@ QueryNetworkResult Engine::QueryNetwork(const CNNNetwork& network, const std::ma
}
auto clonedNetwork = InferenceEngine::details::cloneNetwork(network);
auto ops = clonedNetwork.getFunction()->get_ordered_ops();
const auto& lptProp = config.find(InferenceEngine::PluginConfigInternalParams::KEY_LP_TRANSFORMS_MODE);
const bool enableLPT = (lptProp != config.end() && lptProp->second == PluginConfigParams::YES) /* enabled in the orig_config*/
|| Config::LPTransformsMode::On == engConfig.lpTransformsMode /* or already enabled */;
Transformation(clonedNetwork, enableLPT);
auto ops = clonedNetwork.getFunction()->get_ordered_ops();
std::unordered_set<std::string> supported;
std::unordered_set<std::string> unsupported;
for (auto op : ops) {

4
inference-engine/src/transformations/src/transformations/common_optimizations/nop_elimination.cpp
View File

@ -148,8 +148,7 @@ static bool replace_squeeze_unsqueeze(const std::shared_ptr<Node>& node) {
shared_ptr<Node> reshape;
auto input = node->input_value(0).get_node_shared_ptr();
auto pat =
opset3::Constant::create<int64_t>(element::i64, Shape{target_shape.size()}, target_shape);
auto pat = opset3::Constant::create<int64_t>(element::i64, Shape{target_shape.size()}, target_shape);
if (ov::is_type<opset3::Reshape>(input) || ov::is_type<opset3::Squeeze>(input) ||
ov::is_type<opset3::Unsqueeze>(input)) {
@ -160,6 +159,7 @@ static bool replace_squeeze_unsqueeze(const std::shared_ptr<Node>& node) {
// skip if reshape is nop
if (reshape->get_input_partial_shape(0).same_scheme(shape_ps)) {
copy_runtime_info({input, node->output(0).get_node_shared_ptr()}, node->output(0).get_node_shared_ptr());
return replace_output_update_name(node->output(0), reshape->input_value(0));
} else {
return replace_node_update_name(node, reshape);