diff --git a/ge/graph/passes/dynamic_single_op_reset_shape_pass.h b/ge/graph/passes/dynamic_single_op_reset_shape_pass.h
index 50ec13ba..659bed9c 100644
--- a/ge/graph/passes/dynamic_single_op_reset_shape_pass.h
+++ b/ge/graph/passes/dynamic_single_op_reset_shape_pass.h
@@ -28,7 +28,7 @@ class DynamicSingleOpResetShapePass : public GraphPass {
  private:
   Status ResetOpShape(OpDescPtr &op_desc);
   Status CheckAllAicpuNodes(const ComputeGraphPtr &graph, bool &is_not_aicpu);
-  bool CheckIfConstInput();
+  bool CheckIfConstInput(const GeTensorDescPtr &input_tensor_desc);
 };
 }  // namespace ge
 #endif  // GE_GRAPH_PASSES_DYNAMIC_SINGLE_OP_RESET_SHAPE_PASS_H_