From 07f2ca658858343adf826fc62fcd6bd6f43f2c87 Mon Sep 17 00:00:00 2001 From: dabaiji Date: Tue, 7 Mar 2023 15:20:30 +0800 Subject: [PATCH] fix stub tensor predict output type --- mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h | 3 --- 1 file changed, 3 deletions(-) diff --git a/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h b/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h index bf6d2c463b3..c5d53bcbc18 100644 --- a/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h +++ b/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h @@ -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},