昇腾社区首页
中文
注册

CloneOperationParam

功能说明

拷贝Operation的Param参数到第二个入参。

定义

1
Status CloneOperationParam(const Operation *operation, OpParam &opParam);

参数列表

参数名称

描述

operation

Operation指针。

opParam

OpParam的引用,将返回operation的opParam浅拷贝。

返回值说明

类型

输入/输出

说明

Status

输出

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

约束说明

支持atb/train_op_params.h的参数和EventOperation

使用示例

  1. 创建Operation。
    1
    2
    3
    4
    5
    train::FastSoftMaxParam param;
    param.headNum = 3;
    param.qSeqLen = { 10, 20, 30 };
    atb::Operation *op = nullptr;
    atb::Status st = atb::CreateOperation(param, &op);
    
  2. 获取当前Operation的param。
    1
    2
    train::FastSoftMaxParam cloneParam;
    st = atb::CloneOperationParam(op, cloneParam);