Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix stub tensor predict output type

This commit is contained in:
dabaiji 2023-03-07 15:20:30 +08:00
parent f2c7cbb3bc
commit 07f2ca6588
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h
View File

@ -55,10 +55,7 @@ inline static PredictOutTypeMap out_type_prediction = {{"ActsULQ", kTupleTensor4
{"ApplyAdagradV2", kTupleTensor2},
{"ApplyAdamWithAmsgrad", kTupleTensor4},
{"ApplyAddSign", kTupleTensor2},
{"ApplyCenteredRMSProp", kTupleTensor4},
{"ApplyFtrl", kTupleTensor3},
{"ApplyKerasMomentum", kTupleTensor2},
{"ApplyMomentum", kTupleTensor2},
{"ApplyPowerSign", kTupleTensor2},
{"ApplyProximalAdagrad", kTupleTensor2},
{"ApplyRMSProp", kTupleTensor3},