CloneOperationParam
函数功能
拷贝Operation的Param到第二个入参。
函数原型
Status CloneOperationParam(const Operation *operation, OpParam &opParam)
约束
只支持atb/train_op_params.h的参数。
参数说明
参数名称 |
描述 |
|---|---|
operation |
Operation指针。 |
opParam |
OpParam的引用，将返回operation的opParam浅拷贝。 |
返回值说明
类型 |
输入/输出 |
说明 |
|---|---|---|
Status |
输出 |
状态值，如果成功，返回NO_ERROR。 |
使用样例
- 创建Operation。
train::FastSoftMaxParam param; param.headNum = 3; param.qSeqLen = { 10, 20, 30 }; atb::Operation *op = nullptr; atb::Status st = atb::CreateOperation(param, &op); - 获取当前Operation的param。
train::FastSoftMaxParam cloneParam; st = atb::CloneOperationParam(op, cloneParam);
父主题： atb/operation.h