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_
#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);



Loading…
Cancel
Save