From: @yeyunpeng2020 Reviewed-by: @zhang_xue_tong Signed-off-by: @zhang_xue_tongtags/v1.2.0-rc1
| @@ -42,6 +42,9 @@ int ReshapeBaseCPUKernel::RunImpl(int task_id) { | |||||
| size_t start_index = task_id * cal_max_num_per_thread_; | size_t start_index = task_id * cal_max_num_per_thread_; | ||||
| auto cur_in_ptr = input_ptr_ + start_index; | auto cur_in_ptr = input_ptr_ + start_index; | ||||
| auto cur_out_ptr = output_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; | 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; | 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); | memcpy(cur_out_ptr, cur_in_ptr, data_size); | ||||
| @@ -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_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_; | uint8_t *cur_out_ptr = output_ptr_ + caled_num * out_shape[split_axis_] * inner_size_; | ||||
| int cur_outer = outer_ - caled_num; | int cur_outer = outer_ - caled_num; | ||||
| if (cur_outer <= 0) { | |||||
| return RET_OK; | |||||
| } | |||||
| if (cur_outer > cal_num_per_thread_) { | if (cur_outer > cal_num_per_thread_) { | ||||
| 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_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_; | uint8_t *cur_out_ptr = output_ptr_ + caled_num * inner_size_; | ||||
| int cal_axis_num = out_shape[split_axis_] - caled_num; | 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_) { | if (cal_axis_num > cal_num_per_thread_) { | ||||
| cal_axis_num = cal_num_per_thread_; | cal_axis_num = cal_num_per_thread_; | ||||
| } | } | ||||
| @@ -71,3 +71,5 @@ ml_text_division | |||||
| ml_video_edit_person_divison | ml_video_edit_person_divison | ||||
| ml_video_edit_img_segment | ml_video_edit_img_segment | ||||
| ml_video_edit_video_segment_gauss_adaptis_part1 | ml_video_edit_video_segment_gauss_adaptis_part1 | ||||
| ml_video_edit_Mnet | |||||
| ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 | |||||
| @@ -66,3 +66,5 @@ PoseNet_dla_17_x512 5 | |||||
| ml_location_scene_division 8 | ml_location_scene_division 8 | ||||
| ml_tabel_recog 0.1 | ml_tabel_recog 0.1 | ||||
| ml_text_division 12 | ml_text_division 12 | ||||
| ml_video_edit_Mnet 11 # Further analysis in the future | |||||
| ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5 | |||||
| @@ -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_gongnongbing.onnx 10 | ||||
| ml_video_edit_style_transfer_starry.onnx 11 | ml_video_edit_style_transfer_starry.onnx 11 | ||||
| porseg_tmp.onnx 1 2 | porseg_tmp.onnx 1 2 | ||||
| ml_video_edit_Mnet 1.5 | |||||
| ml_video_edit_hairSeg_have_imageProcessLayer_interpTo145 0.5 | |||||