|
|
|
@@ -17,9 +17,9 @@ |
|
|
|
#ifndef 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 "cce/aicpu_engine_struct.h" |
|
|
|
#include "hybrid/node_executor/node_executor.h" |
|
|
|
#include "aicpu_ext_info.h" |
|
|
|
|
|
|
|
namespace ge { |
|
|
|
@@ -51,7 +51,7 @@ class AicpuNodeTaskBase : public NodeTask { |
|
|
|
|
|
|
|
virtual Status UpdateOutputShapeFromExtInfo(TaskContext &task_context); |
|
|
|
|
|
|
|
Status UpdateShapeAndDataByResultSummary(TaskContext &context); |
|
|
|
virtual Status UpdateShapeAndDataByResultSummary(TaskContext &context); |
|
|
|
|
|
|
|
Status InitForDependComputeTask(bool is_tfkernel); |
|
|
|
|
|
|
|
|