UpdateOperationParam

函数功能

更新Operation的Param参数。

函数原型

Status UpdateOperationParam(Operation *operation, const OpParam &opParam)

约束

只支持atb/train_op_params.h的参数。

参数说明

返回值说明

使用样例

1. 创建Operation。

   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。

   train::FastSoftMaxParam cloneParam;
   st = atb::CloneOperationParam(op, cloneParam);

3. 更新当前Operation的param。

   cloneParam.headNum = 66;
   st = atb::UpdateOperationParam(op, cloneParam);