Home
last modified time | relevance | path

Searched refs:quant_param_ (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dsub_int8.cc28 if (quant_param_ != nullptr) { in ~SubInt8CPUKernel()
29 free(quant_param_); in ~SubInt8CPUKernel()
30 quant_param_ = nullptr; in ~SubInt8CPUKernel()
44 quant_param_ = reinterpret_cast<SubQuantArg *>(malloc(sizeof(SubQuantArg))); in Init()
45 if (quant_param_ == nullptr) { in Init()
49 quant_param_->in0_args_.scale_ = input0->quant_params().front().scale; in Init()
50 quant_param_->in0_args_.zp_ = -input0->quant_params().front().zeroPoint; in Init()
51 quant_param_->in1_args_.scale_ = input1->quant_params().front().scale; in Init()
52 quant_param_->in1_args_.zp_ = -input1->quant_params().front().zeroPoint; in Init()
53 quant_param_->out_args_.scale_ = output->quant_params().front().scale; in Init()
[all …]
Dmatmul_base_int8.cc44 …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()
51 … cur_oc, param_->deep_16_, input_sums_, weight_bias_sums_ + cur_stride, quant_param_->out_act_min_, in RunImpl()
52 … quant_param_->out_act_max_, quant_param_->output_.zp_, cur_mul, cur_left, cur_right, param_->col_, in RunImpl()
70 if (quant_param_ != nullptr) { in FreeQuantParam()
71 if (quant_param_->filter_scale_ != nullptr) { in FreeQuantParam()
72 free(quant_param_->filter_scale_); in FreeQuantParam()
73 quant_param_->filter_scale_ = nullptr; in FreeQuantParam()
[all …]
Dsoftmax_int8.cc33 if (quant_param_ != nullptr) { in ~SoftmaxInt8CPUKernel()
34 free(quant_param_); in ~SoftmaxInt8CPUKernel()
35 quant_param_ = nullptr; in ~SoftmaxInt8CPUKernel()
44 quant_param_ = reinterpret_cast<SoftmaxQuantArg *>(malloc(sizeof(SoftmaxQuantArg))); in Init()
45 if (quant_param_ == nullptr) { in Init()
54 quant_param_->in_quant_args_.scale_ = in_quant_args.front().scale; in Init()
55 quant_param_->in_quant_args_.zp_ = -in_quant_args.front().zeroPoint; in Init()
61 quant_param_->out_quant_arg_.scale_ = out_quant_args.front().scale; in Init()
62 quant_param_->out_quant_arg_.zp_ = -out_quant_args.front().zeroPoint; in Init()
63 quant_param_->output_activation_min_ = std::numeric_limits<int8_t>::min(); in Init()
[all …]
Dl2_norm_int8.cc27 if (quant_param_ != nullptr) { in ~L2NormInt8CPUKernel()
28 free(quant_param_); in ~L2NormInt8CPUKernel()
29 quant_param_ = nullptr; in ~L2NormInt8CPUKernel()
39 quant_param_ = reinterpret_cast<L2NormQuantArg *>(malloc(sizeof(L2NormQuantArg))); in Init()
40 if (quant_param_ == nullptr) { in Init()
44 quant_param_->in_.scale_ = input->quant_params().front().scale; in Init()
45 quant_param_->in_.zp_ = input->quant_params().front().zeroPoint; in Init()
46 quant_param_->out_.scale_ = output->quant_params().front().scale; in Init()
47 quant_param_->out_.zp_ = output->quant_params().front().zeroPoint; in Init()
89 return L2NormalizationInt8(input_data, output_data, l2_norm_param_, quant_param_, begin, end); in DoExecute()
Dlayer_norm_int8.cc26 if (quant_param_ != nullptr) { in ~LayerNormInt8CPUKernel()
27 free(quant_param_); in ~LayerNormInt8CPUKernel()
28 quant_param_ = nullptr; in ~LayerNormInt8CPUKernel()
46 quant_param_ = reinterpret_cast<LayerNormQuantArg *>(malloc(sizeof(LayerNormQuantArg))); in SetQuantArgs()
47 if (quant_param_ == nullptr) { in SetQuantArgs()
55 quant_param_->in_zp_ = input->quant_params().front().zeroPoint; in SetQuantArgs()
56 quant_param_->in_scale_ = input->quant_params().front().scale; in SetQuantArgs()
62 quant_param_->out_zp_ = output->quant_params().front().zeroPoint; in SetQuantArgs()
63 quant_param_->out_scale_ = output->quant_params().front().scale; in SetQuantArgs()
91 beta_ptr_[i] = src_beta[i] * quant_param_->in_scale_ * gamma_scale; in SetQuantArgs()
[all …]
Ddetection_post_process_int8.cc32 … DoDequantizeInt8ToFp32(data_int8_ + thread_offset, data_fp32_ + thread_offset, quant_param_.scale, in DequantizeInt8ToFp32()
33 quant_param_.zeroPoint, num_unit_thread); in DequantizeInt8ToFp32()
63 quant_param_ = tensor->quant_params().front(); in Dequantize()
Dl2_norm_int8.h36 L2NormQuantArg *quant_param_ = nullptr;
Dsoftmax_int8.h40 SoftmaxQuantArg *quant_param_ = nullptr; variable
Dsub_int8.h41 SubQuantArg *quant_param_ = nullptr;
Ddetection_post_process_int8.h45 lite::LiteQuantParam quant_param_; variable
Dlayer_norm_int8.h47 LayerNormQuantArg *quant_param_ = nullptr; variable
Dmatmul_base_int8.h66 MatmulQuantParameter *quant_param_ = nullptr; variable