diff options
| -rw-r--r-- | ethosu/vela/high_level_command_to_npu_op.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/high_level_command_to_npu_op.py b/ethosu/vela/high_level_command_to_npu_op.py index 31434835..8d6fc871 100644 --- a/ethosu/vela/high_level_command_to_npu_op.py +++ b/ethosu/vela/high_level_command_to_npu_op.py @@ -428,6 +428,8 @@ def create_npu_elementwise_op(cmd: NpuStripe, arch: ArchitectureFeatures) -> Npu # Force output scale same as the input scale for # resizebilinear 1x1 that is converted to add output_scale = npu_op.ifm2.quantization.scale_f32 + if op.type == Op.Abs: + output_scale = npu_op.ifm.quantization.scale_f32 / npu_op.ofm.quantization.scale_f32 if op.type == Op.LeakyRelu: output_scale = op.attrs["alpha"] if op.type == Op.RescaleAdd: |