From 265ccd51907b8bbafb823aa81a20197fd38123f3 Mon Sep 17 00:00:00 2001 From: huangchengnuo Date: Tue, 7 Mar 2023 14:26:44 +0800 Subject: [PATCH] Add broadcast in PredictOutMap --- mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h | 1 + 1 file changed, 1 insertion(+) diff --git a/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h b/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h index bf6d2c463b3..eda4cbd98e5 100644 --- a/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h +++ b/mindspore/ccsrc/pipeline/pynative/predict_out_type_map.h @@ -83,6 +83,7 @@ inline static PredictOutTypeMap out_type_prediction = {{"ActsULQ", kTupleTensor4 {"BatchNormGradWithAddAndActivation", kTupleTensor4}, {"BatchNormGradGrad", kTupleTensor3}, {"BiasDropoutAdd", kTupleTensor2}, + {"Broadcast", kTuple}, {"CSRSparseMatrixToSparseTensor", kTupleTensor3}, {"CTCGreedyDecoder", kTupleTensor4}, {"CTCLoss", kTupleTensor2},