let convolutionbiasbackprop apply to 5D (#2230)

NervanaSystems · Dec 13, 2018 · ada8d9b · ada8d9b
1 parent bf14a38
commit ada8d9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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);