aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/weight_compressor.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
-rw-r--r--ethosu/vela/weight_compressor.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py
index 10a1a6dd..68817035 100644
--- a/ethosu/vela/weight_compressor.py
+++ b/ethosu/vela/weight_compressor.py
@@ -314,7 +314,7 @@ def encode_weight_and_scale_tensor(
# No cache hit, need to perform the encoding
if do_weights:
assert weight_tens.quantization is not None
- assert weight_tens.quantization.scale_f32 is not None
+ assert weight_tens.quantization.scale_f32 is not None or op.explicit_scaling
assert weight_tens.quantization.zero_point is not None
# Early zero-point correction