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

!49896 Add broadcast in PredictOutMap 

Merge pull request !49896 from NaCN/fix_broadcast
This commit is contained in:
i-robot 2023-03-08 04:02:19 +00:00 committed by Gitee
parent f29089c753 265ccd5190
commit c7b145cf3e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
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

@ -79,6 +79,7 @@ inline static PredictOutTypeMap out_type_prediction = {{"ActsULQ", kTupleTensor4
{"BatchNormGradWithAddAndActivation", kTupleTensor4}, {"BatchNormGradWithAddAndActivation", kTupleTensor4},
{"BatchNormGradGrad", kTupleTensor3}, {"BatchNormGradGrad", kTupleTensor3},
{"BiasDropoutAdd", kTupleTensor2}, {"BiasDropoutAdd", kTupleTensor2},
{"Broadcast", kTuple},
{"CSRSparseMatrixToSparseTensor", kTupleTensor3}, {"CSRSparseMatrixToSparseTensor", kTupleTensor3},
{"CTCGreedyDecoder", kTupleTensor4}, {"CTCGreedyDecoder", kTupleTensor4},
{"CTCLoss", kTupleTensor2}, {"CTCLoss", kTupleTensor2},