昇腾社区首页
中文
注册
开发者
下载

Execute

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

头文件/库文件

  • 头文件:#include <graph/custom_op.h>
  • 库文件:待补充

功能说明

Eager类自定义算子的执行函数。

函数原型

1
virtual graphStatus Execute(gert::EagerOpExecutionContext *ctx) = 0

参数说明

参数名

输入/输出

描述

ctx

输入

执行时上下文,可通过上下文获取input tensor,分配输出内存,分配workspace等

返回值说明

参数名

类型

说明

-

graphStatus

GRAPH_SUCCESS(0):成功

其他值:失败

约束说明