Skip to content

Commit

Permalink
roi_pooling.cpp fix
Browse files Browse the repository at this point in the history
  • Loading branch information
allnes committed Oct 23, 2023
1 parent 32d38b3 commit f98bb69
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/plugins/intel_cpu/src/nodes/roi_pooling.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -826,7 +826,8 @@ std::shared_ptr<ROIPooling::ROIPoolingExecutor> ROIPooling::ROIPoolingExecutor::

OV_SWITCH(intel_cpu, ROIPoolingExecutorCreation, ctx, jpp.src_prc,
OV_CASE(Precision::FP32, float),
OV_CASE(Precision::BF16, bfloat16_t))
OV_CASE(Precision::BF16, bfloat16_t),
OV_CASE(Precision::FP16, float16_t))

return ctx.executor;
}
Expand Down

0 comments on commit f98bb69

Please sign in to comment.