Browse Source

fix

pull/1865/head
guopeian 4 years ago
parent
commit
8e4328fd19
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.h

+ 2
- 2
ge/hybrid/node_executor/aicpu/aicpu_node_executor.h View File

@@ -17,9 +17,9 @@
#ifndef GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_ #ifndef GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_
#define GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_ #define GE_HYBRID_KERNEL_AICPU_NODE_EXECUTOR_H_


#include "hybrid/node_executor/node_executor.h"
#include "external/graph/types.h" #include "external/graph/types.h"
#include "cce/aicpu_engine_struct.h" #include "cce/aicpu_engine_struct.h"
#include "hybrid/node_executor/node_executor.h"
#include "aicpu_ext_info.h" #include "aicpu_ext_info.h"


namespace ge { namespace ge {
@@ -51,7 +51,7 @@ class AicpuNodeTaskBase : public NodeTask {


virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context); virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context);


Status UpdateShapeAndDataByResultSummary(TaskContext &context);
virtual Status UpdateShapeAndDataByResultSummary(TaskContext &context);


Status InitForDependComputeTask(bool is_tfkernel); Status InitForDependComputeTask(bool is_tfkernel);




Loading…
Cancel
Save