IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[CPU] Add FP16 support to MatrixNms (#20804)

Browse Source
This commit is contained in:
Aleksandr Voron 2023-11-02 15:40:32 +01:00 committed by GitHub
parent 3f5f923a70
commit e8f21eefae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/plugins/intel_cpu/src/nodes/matrix_nms.cpp
View File

@ -115,7 +115,7 @@ void MatrixNms::initSupportedPrimitiveDescriptors() {
if (!supportedPrimitiveDescriptors.empty())
return;
const std::vector<Precision> supportedFloatPrecision = {Precision::FP32};
const std::vector<Precision> supportedFloatPrecision = {Precision::FP32, Precision::FP16};
const std::vector<Precision> supportedIntOutputPrecision = {Precision::I32, Precision::I64};
checkPrecision(getOriginalInputPrecisionAtPort(NMS_BOXES), supportedFloatPrecision, "boxes", m_inType);

2
src/plugins/intel_cpu/src/nodes/multiclass_nms.cpp
View File

@ -107,7 +107,7 @@ void MultiClassNms::initSupportedPrimitiveDescriptors() {
if (!supportedPrimitiveDescriptors.empty())
return;
const std::vector<Precision> supportedFloatPrecision = {Precision::FP32, Precision::BF16};
const std::vector<Precision> supportedFloatPrecision = {Precision::FP32, Precision::FP16, Precision::BF16};
const std::vector<Precision> supportedIntOutputPrecision = {Precision::I32, Precision::I64};
checkPrecision(getOriginalInputPrecisionAtPort(NMS_BOXES), supportedFloatPrecision, "boxes", m_inType);