atb/train_op_params.h

FastSoftMaxOperation
FastSoftMaxGradOperation
GenAttentionMaskOperation
PadWithHiddenStateOperation
RmsNormBackwardOperation
RopeGradOperation
StridedBatchMatmulOperation
UnpadWithHiddenStateOperation

父主题： API参考