From ff54d36efc1c6d4a146d65f81157127d809ca16b Mon Sep 17 00:00:00 2001
From: l00444296 <lixiwen1@huawei.com>
Date: Thu, 17 Dec 2020 15:41:58 +0800
Subject: [PATCH] Feature: reset shape of dynamic single op

---
 ge/graph/passes/dynamic_single_op_reset_shape_pass.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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_