昇腾社区首页
中文
注册

ApplyAddOutputPass

融合模式

表1中算子增加可选输出和precheck。

表1 支持的算子

算子

output1

output2

output3

precheck

ApplyRMSProp

ms

mom

-

ApplyRmsPropPreCheck

FusedMulApplyMomentumWithoutAccumOut

accum

-

-

FusedMulApplyMomentumPreCheck

FusedMulApplyMomentumExtern

accum

-

-

FusedMulApplyMomentumExternPreCheck

FusedMulApplyKerasMomentum

accum

-

-

FusedMulApplyKerasMomentumPreCheck

ApplyAdagrad

accum

-

-

-

ApplyAdagradDA

gradient_accumulator

gradient_squared_accumulator

-

-

ApplyAdadelta

accum

accum_update

-

-

ApplyPowerSign

m

-

-

-

ApplyProximalAdagrad

accum

-

-

-

ApplyAdaMax

m

v

-

-

ApplyAdagradV2

accum

-

-

ApplyAdagradV2PreCheck

ApplyKerasMomentum

accum

-

-

ApplyKerasMomentumPreCheck

ApplyFtrlV2

accum

linear

-

-

ApplyMomentum

accum

-

-

-

ApplyFtrl

accum

linear

-

-

ApplyAdam

m

v

-

-

ApplyCenteredRMSProp

mg

ms

mom

-

ApplyAddSign

m

-

-

-

ApplyAdamWithAmsgrad

m

v

vhat

-

使用约束

支持的型号

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件