forked from mindspore-Ecosystem/mindspore
!11789 add model to ci
From: @yeyunpeng2020 Reviewed-by: @zhang_xue_tong Signed-off-by: @zhang_xue_tong
This commit is contained in:
commit
91878176da
|
@ -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);
|
||||
|
|
|
@ -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_;
|
||||
}
|
||||
|
|
|
@ -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
|
|
@ -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
|
|
@ -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
|
Loading…
Reference in New Issue