返回顶部 atb/train_op_params.h operator== FastSoftMaxOperation FastSoftMaxGradOperation GenAttentionMaskOperation LaserAttentionOperation LaserAttentionGradOperation PadWithHiddenStateOperation RmsNormBackwardOperation RopeGradOperation StridedBatchMatmulOperation UnpadWithHiddenStateOperation
