Skip to content
This repository was archived by the owner on Jan 3, 2023. It is now read-only.

Commit ada8d9b

Browse files
Matthew BrookhartAdam Procter
authored andcommitted
let convolutionbiasbackprop apply to 5D (#2230)
1 parent bf14a38 commit ada8d9b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/ngraph/runtime/cpu/pass/cpu_assignment.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,8 @@ namespace ngraph
281281
data_dilated = data_dilated || (s != 1);
282282
}
283283

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

0 commit comments

Comments
 (0)