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

[IE CLDNN] Forcing bfzyx format in MVN layer (#1903)

Browse Source
This commit is contained in:
Lukasz Debski 2020-08-24 12:03:21 +02:00 committed by GitHub
parent d7db4974f3
commit 0a5dfc62b2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
inference-engine/thirdparty/clDNN/src/layout_optimizer.cpp vendored
View File

@ -852,6 +852,11 @@ format layout_optimizer::get_preferred_format(program_node& node) {
auto& deconv_node = node.as<deconvolution>(); auto& deconv_node = node.as<deconvolution>();
auto weights_layout = deconv_node.weights(0).get_output_layout(); auto weights_layout = deconv_node.weights(0).get_output_layout();
expected = get_expected_layout(output_layout, deconv_node, weights_layout).format; expected = get_expected_layout(output_layout, deconv_node, weights_layout).format;
} else if (node.is_type<mvn>()) {
auto input_layout = node.get_dependency(0).get_output_layout();
if (input_layout.format.dimension() == 5 &&
(input_layout.data_type == data_types::f32 || input_layout.data_type == data_types::f16 ))
expected = format::bfzyx;
} }
return expected; return expected;