let convolutionbiasbackprop apply to 5D (#2230)

ada8d9b8 · Matthew Brookhart · Adam Procter · bf14a381 · ada8d9b8
Commit ada8d9b8 authored Dec 13, 2018 by Matthew Brookhart Committed by Adam Procter Dec 13, 2018
Hide 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
@@ -281,7 +281,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);