!11789 add model to ci

From: @yeyunpeng2020
Reviewed-by: @zhang_xue_tong
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-01-29 14:17:17 +08:00 committed by Gitee
commit 91878176da
5 changed files with 15 additions and 0 deletions

View File

@ -42,6 +42,9 @@ int ReshapeBaseCPUKernel::RunImpl(int task_id) {
size_t start_index = task_id * cal_max_num_per_thread_;
auto cur_in_ptr = input_ptr_ + start_index;
auto cur_out_ptr = output_ptr_ + start_index;
if (start_index > in_tensors_.front()->Size()) {
return RET_OK;
}
size_t data_size = in_tensors_.front()->Size() - start_index;
data_size = data_size > cal_max_num_per_thread_ ? cal_max_num_per_thread_ : data_size;
memcpy(cur_out_ptr, cur_in_ptr, data_size);

View File

@ -110,6 +110,9 @@ int StridedSliceCPUKernel::FastRunImpl(int task_id) {
uint8_t *cur_in_ptr = input_ptr_ + (caled_num * in_shape[split_axis_] + begin_index) * inner_size_;
uint8_t *cur_out_ptr = output_ptr_ + caled_num * out_shape[split_axis_] * inner_size_;
int cur_outer = outer_ - caled_num;
if (cur_outer <= 0) {
return RET_OK;
}
if (cur_outer > cal_num_per_thread_) {
cur_outer = cal_num_per_thread_;
}
@ -120,6 +123,9 @@ int StridedSliceCPUKernel::FastRunImpl(int task_id) {
uint8_t *cur_in_ptr = input_ptr_ + (caled_num * param_->strides_[split_axis_] + begin_index) * inner_size_;
uint8_t *cur_out_ptr = output_ptr_ + caled_num * inner_size_;
int cal_axis_num = out_shape[split_axis_] - caled_num;
if (cal_axis_num <= 0) {
return RET_OK;
}
if (cal_axis_num > cal_num_per_thread_) {
cal_axis_num = cal_num_per_thread_;
}

View File

@ -71,3 +71,5 @@ ml_text_division
ml_video_edit_person_divison
ml_video_edit_img_segment
ml_video_edit_video_segment_gauss_adaptis_part1
ml_video_edit_Mnet
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145

View File

@ -66,3 +66,5 @@ PoseNet_dla_17_x512 5
ml_location_scene_division 8
ml_tabel_recog 0.1
ml_text_division 12
ml_video_edit_Mnet 11 # Further analysis in the future
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5

View File

@ -32,3 +32,5 @@ ml_video_edit_style_transfer_candy.onnx 11
ml_video_edit_style_transfer_gongnongbing.onnx 10
ml_video_edit_style_transfer_starry.onnx 11
porseg_tmp.onnx 1 2
ml_video_edit_Mnet 1.5
ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5