Home
last modified time | relevance | path

Searched refs:deep_16_ (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dmatmul_base_int8.cc50 …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()
180 param_->deep_16_ = UP_ROUND(param_->deep_, C16NUM); in ResizeParameter()
211 auto current_b_pack = pack_b_ptr_ + i * param_->col_align_ * param_->deep_16_; in TransferB()
235 …pack_a_ptr_ = reinterpret_cast<int8_t *>(malloc(param_->row_align_ * param_->deep_16_ * sizeof(int… in InitTmpBuffer()
241 …reinterpret_cast<int8_t *>(malloc(param_->batch * param_->col_align_ * param_->deep_16_ * sizeof(i… in InitTmpBuffer()
257 memset(pack_a_ptr_, 0, param_->row_align_ * param_->deep_16_ * sizeof(int8_t)); in InitTmpBuffer()
258 memset(pack_b_ptr_, 0, param_->batch * param_->col_align_ * param_->deep_16_ * sizeof(int8_t)); in InitTmpBuffer()
337 batch_b_ptr_ = pack_b_ptr_ + i * param_->col_align_ * param_->deep_16_; in Run()
Dconvolution_1x1_int8.cc344 matmul_param_->deep_16_ = UP_ROUND(matmul_param_->deep_, C16NUM); in InitParam()
431 …t8_t *hw_packed_in = packed_input_ + task_id * thread_stride_hw_ * C4NUM * matmul_param_->deep_16_; in RunArmHw()
437 …utSum16x4PerLayer(hw_packed_in, hw_input_sum, 1, UP_ROUND(cur_hw, C4NUM), matmul_param_->deep_16_); in RunArmHw()
440 UP_ROUND(cur_hw, C4NUM), matmul_param_->deep_16_); in RunArmHw()
444 … matmul_param_->col_, matmul_param_->deep_16_, left_shift_, right_shift_, multiplier_, conv_param_, in RunArmHw()
520 …Conv1x1Int8(packed_input_, packed_weight_ + cur_stride * matmul_param_->deep_16_, output_ptr_ + cu… in RunArmOc()
522 … matmul_param_->deep_16_, cur_left_shift, cur_right_shift, cur_multiplier, conv_param_, cur_zp); in RunArmOc()
570 …InputSum16x4PerLayer(packed_input_, input_sum_, 1, matmul_param_->row_4_, matmul_param_->deep_16_); in Run()
573 matmul_param_->row_4_, matmul_param_->deep_16_); in Run()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dmatmul_base_int8_coder.cc34 a_pack_ptr_size_ = param_->row_align_ * param_->deep_16_ * sizeof(int8_t); in InitTmpBuffer()
37 b_pack_ptr_size_ = param_->batch * param_->col_align_ * param_->deep_16_ * sizeof(int8_t); in InitTmpBuffer()
63 param_->deep_16_ = UP_ROUND(param_->deep_, C16NUM); in ResizeParameter()
205 … param_->deep_, param_->col_, param_->col_align_, param_->deep_16_, quant_.input_.zp_, in DoCode()
210 … param_->col_, param_->col_align_, param_->deep_16_, quant_.input_.zp_, "init_filter_zp", in DoCode()
232 …batch_b_ptr_str = pack_b_ptr_str + "+" + std::to_string(i * param_->col_align_ * param_->deep_16_); in DoCode()
258 …ng batch_b_ptr_str_final = batch_b_ptr_str + " + " + std::to_string(cur_stride * param_->deep_16_); in DoCode()
262 … param_->deep_16_, input_sums_, weight_bias_sums_str_final, quant_.out_act_min_, in DoCode()
Dconv2d_1x1_int8_coder.cc243 matmul_param_->deep_16_ = UP_ROUND(matmul_param_->deep_, C16NUM); in InitParam()
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv1x1_run_int8_wrapper.c104 …Conv1x1Int8(args->packed_input_, args->packed_weight_ + cur_stride * args->matmul_param_->deep_16_, in RunArmOc()
106 … args->matmul_param_->row_, cur_oc, args->matmul_param_->deep_16_, cur_left_shift, cur_right_shift, in RunArmOc()
149 args->packed_input_ + task_id * args->thread_stride_hw_ * C4NUM * args->matmul_param_->deep_16_; in RunArmHw()
155 …6x4PerLayer(hw_packed_in, hw_input_sum, 1, UP_ROUND(cur_hw, C4NUM), args->matmul_param_->deep_16_); in RunArmHw()
158 UP_ROUND(cur_hw, C4NUM), args->matmul_param_->deep_16_); in RunArmHw()
162 … args->matmul_param_->col_, args->matmul_param_->deep_16_, args->left_shift_, args->right_shift_, in RunArmHw()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dmatmul_parameter.h56 int deep_16_; member
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/
Dnnacl_fp32_serializer.cc80 …rameter.col_align_, mat_mul_parameter.deep_, mat_mul_parameter.deep_4_, mat_mul_parameter.deep_16_, in CodeStruct()
Dnnacl_int8_serializer.cc69 matmul_parameter.deep_, matmul_parameter.deep_4_, matmul_parameter.deep_16_, in CodeStruct()