diff --git a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
index 02180b1d..8980d41b 100644
--- a/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
+++ b/ge/hybrid/node_executor/aicpu/aicpu_node_executor.h
@@ -55,12 +55,11 @@ class AicpuNodeTaskBase : public NodeTask {
 
   virtual Status LaunchTask(TaskContext &context) = 0;
 
-  virtual Status TaskCallback(TaskContext &context) = 0;
-
   virtual Status UpdateIoAddr(TaskContext &context) = 0;
 
   static Status AllocTensorBuffer(size_t size, std::unique_ptr<TensorBuffer> &tensor_buffer);
-
+ private:
+  Status TaskCallback(TaskContext &context);
  protected:
   const NodeItem *node_item_;
   // just reference.
@@ -93,8 +92,6 @@ class AicpuTfNodeTask : public AicpuNodeTaskBase {
 
   Status LaunchTask(TaskContext &context) override;
 
-  Status TaskCallback(TaskContext &context) override;
-
   Status UpdateIoAddr(TaskContext &context) override;
 
  private:
@@ -162,7 +159,7 @@ class AicpuNodeTask : public AicpuNodeTaskBase {
 
   Status LaunchTask(TaskContext &context) override;
 
-  Status TaskCallback(TaskContext &context) override;
+   override;
 
   Status UpdateIoAddr(TaskContext &context) override;
 
@@ -171,6 +168,8 @@ class AicpuNodeTask : public AicpuNodeTaskBase {
 
   Status InitForDependComputeTask();
 
+  Status UpdateShapeAndDataByResultSummary(TaskContext &context);
+
  protected:
   // host mem
   std::unique_ptr<uint8_t[]> args_;