diff options
-rw-r--r-- | src/TosaSerialize.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp index b894383..45a1b37 100644 --- a/src/TosaSerialize.cpp +++ b/src/TosaSerialize.cpp @@ -307,7 +307,7 @@ std::string TosaSerializationOperatorBuilder::GetVariableTensorName( mlir::Operation *op) const { mlir::Attribute variable_op_name_attr = op->getAttr("name"); std::string variable_tensor_mlir_name = - op->getAttr("name").cast<mlir::FlatSymbolRefAttr>().getValue().str(); + op->getAttr("name").cast<mlir::StringAttr>().getValue().str(); if (variable_tensor_flatbuffer_name_map.find(variable_tensor_mlir_name) == variable_tensor_flatbuffer_name_map.end()) { |