diff options
| author | Tai Ly <tai.ly@arm.com> | 2023-04-25 15:36:20 +0000 |
|---|---|---|
| committer | Tai Ly <tai.ly@arm.com> | 2023-04-25 21:47:27 +0000 |
| commit | 828099da4d3777254688f3a75063a705762064d6 (patch) | |
| tree | b93fa42aa35916edb3857a69e13bbad03ad0907d | |
| parent | 67e9fc539dd014745f8e2559b967489b8479a8f8 (diff) | |
| download | tosa_mlir_translator-828099da4d3777254688f3a75063a705762064d6.tar.gz | |
Fix deserialization of PadOp
Pad constant attribute requires scalar values (instead of 1D tensor)
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: Iae0bf49b1b404c7b93d515a10b182c3906054e70
| -rw-r--r-- | src/TosaDeserialize.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/TosaDeserialize.cpp b/src/TosaDeserialize.cpp index d30a6f2..3d38e1b 100644 --- a/src/TosaDeserialize.cpp +++ b/src/TosaDeserialize.cpp @@ -742,7 +742,7 @@ std::vector<mlir::Value> TosaMlirOperatorBuilder::build<Op_PAD>(TosaSerializatio mlir::Value pad_const_value; if (pad_const_int != 0) { auto pad_const_int_type = - mlir::RankedTensorType::get({1}, op_builder->getI32Type()); + mlir::RankedTensorType::get({}, op_builder->getI32Type()); auto pad_const_int_attr = mlir::DenseElementsAttr::get(pad_const_int_type, {pad_const_int}); mlir::Operation *pad_const_int_op = @@ -752,7 +752,7 @@ std::vector<mlir::Value> TosaMlirOperatorBuilder::build<Op_PAD>(TosaSerializatio pad_const_value = pad_const_int_op->getResult(0); } else if (pad_const_fp != 0) { auto pad_const_fp_type = - mlir::RankedTensorType::get({1}, op_builder->getF32Type()); + mlir::RankedTensorType::get({}, op_builder->getF32Type()); auto pad_const_fp_attr = mlir::DenseElementsAttr::get(pad_const_fp_type, {pad_const_fp}); mlir::Operation *pad_const_fp_op = |