diff --git a/register/auto_mapping_util.cpp b/register/auto_mapping_util.cpp
index af3c8e1dae3c2baf16eae949385b25a2cdf44563..b58d52b9ec2f33c5c9a9d9e3af2f056630474b87 100644
--- a/register/auto_mapping_util.cpp
+++ b/register/auto_mapping_util.cpp
@@ -46,7 +46,7 @@ void AutoMappingUtil::ConvertShape(const domi::tensorflow::TensorShapeProto &sha
       shape_dims.push_back(dim.size());
     }
   } else {
-    shape_dims = ge::UNKNOWN_SHAPE;
+    shape_dims = ge::UNKNOWN_RANK;
   }
 }