Home
last modified time | relevance | path

Searched refs:cur_stride (Results 1 – 11 of 11) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv1x1_run_int8_wrapper.c35 int cur_stride = args->thread_stride_hw_ * C4NUM; in OcOptPre() local
37 int cur_hw = MSMIN(cur_stride, res_stride); in OcOptPre()
57 int cur_stride = task_id * stride; in RunArm64OptOc() local
58 int res_stride = args->matmul_param_->col_ - cur_stride; in RunArm64OptOc()
66 filter_peroc ? args->left_shift_ + cur_stride : args->conv_param_->conv_quant_arg_.left_shift_; in RunArm64OptOc()
68 … filter_peroc ? args->right_shift_ + cur_stride : args->conv_param_->conv_quant_arg_.right_shift_; in RunArm64OptOc()
70 …filter_peroc ? args->multiplier_ + cur_stride : args->conv_param_->conv_quant_arg_.quant_multiplie… in RunArm64OptOc()
71 int32_t *cur_zp = filter_peroc ? args->filter_zp_ptr_ + cur_stride : args->filter_zp_ptr_; in RunArm64OptOc()
73 …Conv1x1Int8Opt(args->packed_input_, args->packed_weight_ + cur_stride * args->matmul_param_->deep_… in RunArm64OptOc()
74 args->output_ptr_ + cur_stride, args->input_sum_, args->bias_data_ + cur_stride, in RunArm64OptOc()
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconvolution_1x1_int8.cc422 int cur_stride = thread_stride_hw_ * C4NUM; in RunArmHw() local
424 int cur_hw = MSMIN(cur_stride, res_stride); in RunArmHw()
450 int cur_stride = thread_stride_hw_ * C4NUM; in RunArm64OptHw() local
452 int cur_hw = MSMIN(cur_stride, res_stride); in RunArm64OptHw()
477 int cur_stride = task_id * stride; in RunArm64OptOc() local
478 int res_stride = matmul_param_->col_ - cur_stride; in RunArm64OptOc()
484 …int32_t *cur_left_shift = filter_peroc_ ? left_shift_ + cur_stride : conv_param_->conv_quant_arg_.… in RunArm64OptOc()
485 …int32_t *cur_right_shift = filter_peroc_ ? right_shift_ + cur_stride : conv_param_->conv_quant_arg… in RunArm64OptOc()
486 …int32_t *cur_multiplier = filter_peroc_ ? multiplier_ + cur_stride : conv_param_->conv_quant_arg_.… in RunArm64OptOc()
487 int32_t *cur_zp = filter_peroc_ ? filter_zp_ptr_ + cur_stride : filter_zp_ptr_; in RunArm64OptOc()
[all …]
Dmatmul_base_int8.cc37 int cur_stride = task_id * stride; in RunImpl() local
38 int res_stride = param_->col_ - cur_stride; in RunImpl()
44 …int32_t *cur_left = filter_per_channel_ ? quant_param_->left_shift_ + cur_stride : quant_param_->l… in RunImpl()
45 …int32_t *cur_right = filter_per_channel_ ? quant_param_->right_shift_ + cur_stride : quant_param_-… in RunImpl()
47 …filter_per_channel_ ? quant_param_->quant_multiplier_ + cur_stride : quant_param_->quant_multiplie… in RunImpl()
48 …int32_t *cur_zp = filter_per_channel_ ? quant_param_->filter_zp_ + cur_stride : quant_param_->filt… in RunImpl()
50 …MatmulInt8Opt(pack_a_ptr_, batch_b_ptr_ + cur_stride * param_->deep_16_, batch_c_ptr_ + cur_stride in RunImpl()
51 … cur_oc, param_->deep_16_, input_sums_, weight_bias_sums_ + cur_stride, quant_param_->out_act_min_, in RunImpl()
Ddeconvolution_int8.cc247 int cur_stride = thread_stride_; in DoDeconv() local
249 int cur_oc = MSMIN(cur_stride, res_stride); in DoDeconv()
251 cur_stride = thread_stride_ * C4NUM; in DoDeconv()
253 int cur_oc_res = MSMIN(cur_stride, res_stride); in DoDeconv()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dmatmul_base_int8_coder.cc236 int cur_stride = kDefaultTaskId * stride; in DoCode() local
237 int res_stride = param_->col_ - cur_stride; in DoCode()
248 code << cur_left << " + " << cur_stride << value_str_end; in DoCode()
249 code << cur_right << " + " << cur_stride << value_str_end; in DoCode()
250 code << cur_mul << " + " << cur_stride << value_str_end; in DoCode()
251 code << cur_zp << " + " << cur_stride << value_str_end; in DoCode()
258 …std::string batch_b_ptr_str_final = batch_b_ptr_str + " + " + std::to_string(cur_stride * param_->… in DoCode()
259 std::string batch_c_ptr_final = batch_c_ptr_str + "+" + std::to_string(cur_stride); in DoCode()
260 … std::string weight_bias_sums_str_final = weight_bias_sums_str + "+" + std::to_string(cur_stride); in DoCode()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc161 int cur_stride = UP_DIV(conv_param_->output_channel_, C8NUM) - task_id * thread_stride_; in DoDeconv() local
162 int oc = MSMIN(thread_stride_, cur_stride); in DoDeconv()
163 cur_stride = conv_param_->output_channel_ - task_id * thread_stride_ * C8NUM; in DoDeconv()
164 int oc_res = MSMIN(thread_stride_ * C8NUM, cur_stride); in DoDeconv()
Dconvolution_1x1_fp16.cc200 int cur_stride = matmul_param_->col_ - task_id * thread_stride_; in RunOc() local
201 int cur_oc = MSMIN(thread_stride_, cur_stride); in RunOc()
Dmatmul_base_fp16.cc286 int cur_stride = params_->col_ - task_id * thread_stride_; in RunImpl() local
287 int cur_oc = MSMIN(thread_stride_, cur_stride); in RunImpl()
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/
Ddeconvolution_fp32_wrapper.c28 int cur_stride = thread_stride * C8NUM; in DoDeconvFp32() local
30 int oc_res = MSMIN(cur_stride, res_stride); in DoDeconvFp32()
/third_party/gstreamer/gstplugins_bad/sys/d3d11/
Dgstd3d11pluginutils.cpp640 guint cur_stride = 0; in gst_d3d11_allocate_staging_buffer_for() local
652 if (!gst_d3d11_memory_get_texture_stride (new_mem, &cur_stride) || in gst_d3d11_allocate_staging_buffer_for()
653 cur_stride < staging_desc.Width) { in gst_d3d11_allocate_staging_buffer_for()
660 stride[i] = cur_stride; in gst_d3d11_allocate_staging_buffer_for()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32.cc133 int cur_stride = thread_stride_ * C8NUM; in DoDeconv() local
135 int oc_res = MSMIN(cur_stride, res_stride); in DoDeconv()