diff options
Diffstat (limited to 'src/TosaSerialize.cpp')
-rw-r--r-- | src/TosaSerialize.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index fec9f17..66d0a31 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -931,7 +931,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::PadOp>( paddings.push_back(val); } - int32_t pad_const_int = 0; + auto quant_info = pad_op.getQuantizationInfoAttr(); + // pad_const includes the zero point if the tensor uses a zero point. + int32_t pad_const_int = quant_info ? quant_info.getInputZp() : 0; float pad_const_fp = 0.f; if (auto tensor = pad_op.getPadConst()) { |