enable fp16 convert tests
This commit is contained in:
1 parent
3db3d4294e
commit
037af672ce
2 files changed
+3
-2
No files matched your search
@@ -45,7 +45,7 @@ public:
|
||||
InferenceEngine::Precision::FP16,
|
||||
InferenceEngine::Precision::I32,
|
||||
InferenceEngine::Precision::FP32)) {
|
||||
DEBUG_LOG("NECopy does not support source precision: ", convertParams.srcPrc.name());
|
||||
DEBUG_LOG("NECast does not support source precision: ", convertParams.srcPrc.name());
|
||||
return false;
|
||||
}
|
||||
if ((convertParams.srcPrc == InferenceEngine::Precision::I8 &&
|
||||
@@ -80,7 +80,7 @@ public:
|
||||
!one_of(convertParams.dstPrc, InferenceEngine::Precision::BF16,
|
||||
InferenceEngine::Precision::FP16,
|
||||
InferenceEngine::Precision::I32))) {
|
||||
DEBUG_LOG("NECopy does not support passed combination of source and destination precisions. ",
|
||||
DEBUG_LOG("NECast does not support passed combination of source and destination precisions. ",
|
||||
"source precision: ", convertParams.srcPrc.name(), " destination precsion: " , convertParams.dstPrc.name());
|
||||
return false;
|
||||
}
|
||||
|
||||
@@ -167,6 +167,7 @@ const std::vector<Precision>& precisions() {
|
||||
Precision::U8,
|
||||
Precision::I8,
|
||||
Precision::I32,
|
||||
Precision::FP16,
|
||||
Precision::FP32,
|
||||
Precision::BF16
|
||||
};
|
||||
|
||||
Reference in new issue
Block a user