!19482 add broadcastto op parser

Merge pull request !19482 from wangyanling/model
This commit is contained in:
i-robot 2021-07-13 07:00:06 +00:00 committed by Gitee
commit 31aa5e3d69
3 changed files with 4 additions and 3 deletions

View File

@ -97,12 +97,10 @@ bool Model::CheckModelSupport(enum DeviceType device_type, ModelType model_type)
std::vector<MSTensor> Model::GetInputs() {
std::vector<MSTensor> empty;
// std::cout << "Model::GetInputs " << std::endl;
if (impl_ == nullptr) {
MS_LOG(ERROR) << "Model implement is null.";
return empty;
}
// std::cout << "Model2::GetInputs " << std::endl;
return impl_->GetInputs();
}

View File

@ -93,3 +93,5 @@ ml_asr_decoder_202103.onnx;2;1,64,512:1,64
decoder.onnx;2;1,7,512:1,7
ml_video_edit_makeup_mobilenetv203.onnx
ml_video_edit_hair_dyeing_migrate_v2.onnx;4
# cur acc for ml_audio_kit_vocals_test is 1.7% because the softmax's output of the last op has very small numbers.
ml_audio_kit_vocals_test.onnx;1;1,512,1024,2 2

View File

@ -99,5 +99,6 @@ hiai_nlu_model.pb;3;1,16:1,16:1,16
gts_object_detect_Ics.pb;1;420,630,3;;input_dependent
hiai_transformer_encoder.pb;15
decoder_step_nocumsum_v5.pb;13;1:1,512:1,1429,2:1,127:1,127:1,127:1,127,320:1,80:1,512:1,512:1,512:1,512:1,512
hiai_nlu_model_v2.pb;7;1,5:1,6:1,174:1,98:1,5:1,5:1,5
ml_audio_kit_encoder_v5.pb;6;1,32:1,32:1,32:1,32:1:1
hiai_nlu_model_v1.pb;3;1,16:1,16:1,16 2.0
hiai_nlu_model_v2.pb;7;1,5:1,6:1,174:1,98:1,5:1,5:1,5