ctcgreedydecoderlen fix

allnes · Oct 12, 2023 · bb809f6 · bb809f6
1 parent 57741d3
commit bb809f6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/plugins/intel_cpu/src/nodes/ctc_greedy_decoder_seq_len.cpp b/src/plugins/intel_cpu/src/nodes/ctc_greedy_decoder_seq_len.cpp
@@ -55,7 +55,7 @@ void CTCGreedyDecoderSeqLen::initSupportedPrimitiveDescriptors() {
         return;
 
     Precision inDataPrecision = getOriginalInputPrecisionAtPort(DATA_INDEX);
-    if (inDataPrecision != Precision::FP32 && inDataPrecision != Precision::BF16 && inDataPrecision != Precision::FP16)
+    if (!one_of(inDataPrecision, Precision::FP32, Precision::BF16, Precision::FP16))
         IE_THROW() << errorPrefix << "has unsupported 'data' input precision: " << inDataPrecision;
 
     Precision seqLenPrecision = getOriginalInputPrecisionAtPort(SEQUENCE_LENGTH_INDEX);