diff options
-rw-r--r-- | src/TosaSerialize.cpp | 47 | ||||
m--------- | third_party/serialization_lib | 0 |
2 files changed, 14 insertions, 33 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index a6ea5ff..7256385 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -968,23 +968,15 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::ResizeOp>( std::string input_name = GetTensorName(op.getOperand(0)); std::string output_name = GetTensorName(op.getResult(0)); - std::vector<int> output_size; - auto output_size_attr = - op.getAttr("output_size").dyn_cast<mlir::ArrayAttr>().getValue(); - for (auto &int_attr : output_size_attr) { - output_size.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt()); + std::vector<int16_t> scale; + auto scale_attr = + op.getAttr("scale").dyn_cast<mlir::ArrayAttr>().getValue(); + for (auto &int_attr : scale_attr) { + scale.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt()); } - ASSERT_VECTOR_LENGTH(output_size, 2); + ASSERT_VECTOR_LENGTH(scale, 4); - std::vector<int> stride; - auto stride_attr = - op.getAttr("stride").dyn_cast<mlir::ArrayAttr>().getValue(); - for (auto &int_attr : stride_attr) { - stride.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt()); - } - ASSERT_VECTOR_LENGTH(stride, 2); - - std::vector<int> offset; + std::vector<int16_t> offset; auto offset_attr = op.getAttr("offset").dyn_cast<mlir::ArrayAttr>().getValue(); for (auto &int_attr : offset_attr) { @@ -992,30 +984,19 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::ResizeOp>( } ASSERT_VECTOR_LENGTH(offset, 2); - int32_t shift = op.getAttr("shift").dyn_cast<mlir::IntegerAttr>().getInt(); - - std::vector<float> stride_fp; - auto stride_fp_attr = - op.getAttr("stride_fp").dyn_cast<mlir::ArrayAttr>().getValue(); - for (auto &fp_attr : stride_fp_attr) { - stride_fp.push_back(fp_attr.dyn_cast<mlir::FloatAttr>().getValueAsDouble()); - } - ASSERT_VECTOR_LENGTH(stride_fp, 2); - - std::vector<float> offset_fp; - auto offset_fp_attr = - op.getAttr("offset_fp").dyn_cast<mlir::ArrayAttr>().getValue(); - for (auto &fp_attr : offset_fp_attr) { - offset_fp.push_back(fp_attr.dyn_cast<mlir::FloatAttr>().getValueAsDouble()); + std::vector<int16_t> border; + auto border_attr = + op.getAttr("border").dyn_cast<mlir::ArrayAttr>().getValue(); + for (auto &int_attr : border_attr) { + border.push_back(int_attr.dyn_cast<mlir::IntegerAttr>().getInt()); } - ASSERT_VECTOR_LENGTH(offset_fp, 2); + ASSERT_VECTOR_LENGTH(border, 2); auto mode_str = op.getAttr("mode").dyn_cast<mlir::StringAttr>().getValue().str(); ResizeMode mode = ResizeModeStr2Enum(mode_str); - TosaResizeAttribute attribute(output_size, stride, offset, shift, stride_fp, - offset_fp, mode); + TosaResizeAttribute attribute(scale, offset, border, mode); TosaSerializationOperator *tyop = new TosaSerializationOperator( Op_RESIZE, Attribute_ResizeAttribute, &attribute, diff --git a/third_party/serialization_lib b/third_party/serialization_lib -Subproject 343d6a703c3a270a01102ec468b59ef2967b595 +Subproject 49b1ca63975d4f1865ca312ed9c9b5e0008e13e |