Searched refs:count_per_thread (Results 1 – 5 of 5) sorted by relevance
45 int count_per_thread = UP_DIV(elements_num_, op_parameter_->thread_num_); in AddNParallelRun() local46 int count = MSMIN(count_per_thread, elements_num_ - thread_id * count_per_thread); in AddNParallelRun()47 auto stride = count_per_thread * thread_id; in AddNParallelRun()
49 int count_per_thread = UP_DIV(elements_num_, op_parameter_->thread_num_); in AddNParallelRun() local50 int count = MSMIN(count_per_thread, elements_num_ - thread_id * count_per_thread); in AddNParallelRun()51 auto stride = count_per_thread * thread_id; in AddNParallelRun()
50 size_t count_per_thread = UP_DIV(out_shape1, crop_param->op_parameter_.thread_num_); in Crop4D() local51 size_t thread_stride = thread_id * count_per_thread; in Crop4D()55 for (size_t j = 0; j < count_per_thread; ++j) { in Crop4D()
35 int count_per_thread = UP_DIV(param->size_[5], param->op_parameter_.thread_num_); in SliceInt8() local36 size_t thread_begin = thread_id * count_per_thread; in SliceInt8()37 size_t thread_end = MSMIN(param->size_[5], thread_begin + count_per_thread); in SliceInt8()
56 int count_per_thread = UP_DIV(param->size_[5], param->op_parameter_.thread_num_); in DoSlice() local57 int thread_begin = thread_id * count_per_thread; in DoSlice()58 int thread_end = MSMIN(param->size_[5], thread_begin + count_per_thread); in DoSlice()