summaryrefslogtreecommitdiff
path: root/compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp')
-rw-r--r--compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp7
1 files changed, 4 insertions, 3 deletions
diff --git a/compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp b/compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp
index c9b35e0be..ef047d35d 100644
--- a/compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp
+++ b/compiler/luci/pass/src/QuantizeDequantizeWeightsPass.cpp
@@ -27,6 +27,7 @@
#include <iostream>
#include <cmath>
#include <functional>
+#include <limits>
namespace
{
@@ -352,15 +353,15 @@ private:
private:
// Check if
// 1. node is const
- // 2. node was not quantized
+ // 2. node's dtype is float32
bool is_quantizable(loco::Node *node)
{
auto const_node = dynamic_cast<luci::CircleConst *>(node);
if (not const_node)
return false;
- // Skip if this is already quantized
- if (is_quantized(const_node))
+ // Skip if this is not float32
+ if (const_node->dtype() != loco::DataType::FLOAT32)
return false;
return true;