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

Add broadcast in PredictOutMap

This commit is contained in:
huangchengnuo 2023-03-07 14:26:44 +08:00
parent 661ec54dde
commit 265ccd5190
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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},