diff options
-rw-r--r-- | src/TosaSerialize.cpp | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index ae5e9c6..8435fe9 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -992,8 +992,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TransposeOp>( return nullptr; std::vector<int> perm; - for (int32_t i = 0; i < perm_elems.getNumElements(); i++) { - perm.push_back(perm_elems.getValue<mlir::IntegerAttr>(i).getInt()); + auto values = perm_elems.getValues<mlir::IntegerAttr>(); + for (int32_t i = 0; i < values.size(); i++) { + perm.push_back(values[i].getInt()); } TosaTransposeAttribute attribute(perm); @@ -1230,8 +1231,9 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::TableOp>( return nullptr; std::vector<int> table; - for (int32_t i = 0; i < table_elems.getNumElements(); i++) { - table.push_back(table_elems.getValue<mlir::IntegerAttr>(i).getInt()); + auto values = table_elems.getValues<mlir::IntegerAttr>(); + for (int32_t i = 0; i < values.size(); i++) { + table.push_back(values[i].getInt()); } TosaTableAttribute attribute(table); |