aboutsummaryrefslogtreecommitdiff
path: root/Utils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'Utils.cpp')
-rw-r--r--Utils.cpp47
1 files changed, 47 insertions, 0 deletions
diff --git a/Utils.cpp b/Utils.cpp
index 873dce4a..60e7a80c 100644
--- a/Utils.cpp
+++ b/Utils.cpp
@@ -603,6 +603,53 @@ bool AreDynamicTensorsSupported()
#endif
}
+bool isQuantizedOperand(const V1_0::OperandType& operandType)
+{
+ if (operandType == V1_0::OperandType::TENSOR_QUANT8_ASYMM)
+ {
+ return true;
+ }
+ else
+ {
+ return false;
+ }
+}
+
+#if defined(ARMNN_ANDROID_NN_V1_2) || defined(ARMNN_ANDROID_NN_V1_3)// Using ::android::hardware::neuralnetworks::V1_2
+bool isQuantizedOperand(const V1_2::OperandType& operandType)
+{
+ if (operandType == V1_2::OperandType::TENSOR_QUANT8_ASYMM ||
+ operandType == V1_2::OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL ||
+ operandType == V1_2::OperandType::TENSOR_QUANT8_SYMM ||
+ operandType == V1_2::OperandType::TENSOR_QUANT16_SYMM )
+ {
+ return true;
+ }
+ else
+ {
+ return false;
+ }
+}
+#endif
+
+#ifdef ARMNN_ANDROID_NN_V1_3 // Using ::android::hardware::neuralnetworks::V1_3
+bool isQuantizedOperand(const V1_3::OperandType& operandType)
+{
+ if (operandType == V1_3::OperandType::TENSOR_QUANT8_ASYMM ||
+ operandType == V1_3::OperandType::TENSOR_QUANT8_SYMM_PER_CHANNEL ||
+ operandType == V1_3::OperandType::TENSOR_QUANT8_SYMM ||
+ operandType == V1_3::OperandType::TENSOR_QUANT16_SYMM ||
+ operandType == V1_3::OperandType::TENSOR_QUANT8_ASYMM_SIGNED)
+ {
+ return true;
+ }
+ else
+ {
+ return false;
+ }
+}
+#endif
+
std::string GetFileTimestamp()
{
// used to get a timestamp to name diagnostic files (the ArmNN serialized graph