昇腾社区首页
中文
注册

AddOperation

功能说明

  • 向图中添加算子。
  • 创建并向图中添加算子。

定义

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
virtual Status AddOperation (Operation *operation, const SVector< std::string > &inTensorNames, const SVector< std::string > &outTensorNames)=0;

template <class OpParam>
Status AddOperation(const OpParam &opParam, const SVector<std::string> &inTensorNames, const SVector< std::string > &outTensorNames)
{
    Operation *operation = nullptr;
    Status st = CreateOperation(opParam, &operation);
    if (st != NO_ERROR) {
        return st;
    }
    st =  AddOperation(operation, inTensorNames, outTensorNames);
    if (st != NO_ERROR) {
        if (operation != nullptr) {
            DestroyOperation(operation);
        }
    }
    return st;
}

成员

成员名称

描述

opParam

要添加的算子参数。

inTensorNames

被添加算子的输入tensor名称。

outTensorNames

被添加算子的输出tensor名称。

返回值

状态值。如果设置成功,返回NO_ERROR。