let convolutionbiasbackprop apply to 5D (#2230) (#2231)

4038c555 · Adam Procter · Scott Cyphers · 870b9b0d · 4038c555
Commit 4038c555 authored Dec 15, 2018 by Adam Procter Committed by Scott Cyphers Dec 15, 2018
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

cpu_assignment.cpp src/ngraph/runtime/cpu/pass/cpu_assignment.cpp +2 -1

No files found.
--- a/src/ngraph/runtime/cpu/pass/cpu_assignment.cpp
+++ b/src/ngraph/runtime/cpu/pass/cpu_assignment.cpp
@@ -292,7 +292,8 @@ namespace ngraph
                        data_dilated = data_dilated || (s != 1);
                    }

-                    if (!data_dilated && data_rank == 4 && delta_rank == 4 &&
+                    if (!data_dilated && data_rank == delta_rank &&
+                        (data_rank == 4 || data_rank == 5) &&
                        node->get_input_element_type(0) == element::f32)
                    {
                        runtime::cpu::mkldnn_utils::assign_mkldnn_kernel(node);