Home
last modified time | relevance | path

Searched refs:output_scale (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/
Dconv2d_base_coder.cc31 auto output_scale = static_cast<double>(conv_quant_arg_->output_quant_args_[0].scale_); in SetQuantArgs() local
40 MS_CHECK_TRUE(output_scale > 0, "output_scale should larger than 0."); in SetQuantArgs()
41 const double effective_output_scale = input_scale * filter_scale / output_scale; in SetQuantArgs()
49 MS_CHECK_TRUE(output_scale > 0, "output_scale should larger than 0."); in SetQuantArgs()
50 const double effective_output_scale = input_scale * filter_scale / output_scale; in SetQuantArgs()
Dmul_int8_coder.cc42 const double output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
44 const double real_multiplier = input1_scale * input2_scale / output_scale; in Prepare()
48 CalculateActivationRangeQuantized(false, false, out_offset_, output_scale, &out_activation_min_, in Prepare()
Dadd_int8_coder.cc46 const double output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
50 MS_CHECK_TRUE(output_scale > 0, "output_scale should larger than 0."); in Prepare()
54 …ce_max_input_scale / ((1 << static_cast<size_t>(left_shift_)) * static_cast<double>(output_scale)); in Prepare()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dsigmoid_int8_coder.cc28 …TableList(int8_t *table, const float input_scale, const int32_t input_zp, const float output_scale, in CalculateTableList() argument
36 const int32_t quantized = std::round(sigmoid_value / output_scale) + output_zp; in CalculateTableList()
49 const float output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
51 CalculateTableList(table_list_, input_scale, input_zp, output_scale, output_zp); in Prepare()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dunsqueeze_int8.c23 float output_scale = para_->quant_arg.out_quant_args_.scale_; in Int8Unsqueeze() local
24 NNACL_CHECK_ZERO_RETURN_ERR(output_scale); in Int8Unsqueeze()
30 output_ptr[i] = output_zp + round(1 / output_scale * input_scale * (input_ptr[i] - input_zp)); in Int8Unsqueeze()
Dpower_int8.c22 double output_scale = param->quant_arg_.out_args_.scale_; in PowerInt8() local
31 int32_t output_scaled = round(output_val / output_scale) + output_zp; in PowerInt8()
41 int32_t output_scaled = round(output_val / output_scale) + output_zp; in PowerInt8()
Dconcat_int8.c23 float output_scale = para->quant_arg_.out_args_.scale_; in Int8Concat() local
24 const float output_inverse_scale = 1.f / output_scale; in Int8Concat()
43 if (input_quant[i].scale_ == output_scale && input_quant[i].zp_ == output_zp) { in Int8Concat()
Dsqueeze_int8.c21 float output_scale = quant_Squeeze_parm->out_quant_args_->scale_; in SqueezeInt8() local
22 const float output_inverse_scale = 1.f / output_scale; in SqueezeInt8()
Dleaky_relu_int8.c24 float output_scale = quant_prelu_parm->quant_arg.out_args_.scale_; in DoLeakReluInt8() local
26 const float output_inverse_scale = 1.f / output_scale; in DoLeakReluInt8()
Dslice_int8.c23 double output_scale = param->quant_arg_.out_args_.scale_; in SliceInt8() local
47 if (input_scale == output_scale && input_zp == output_zp) { in SliceInt8()
Dhswish_int8.h27 double output_scale; member
Dpooling_int8.c37 float output_scale = pooling_param->quant_args_[1][0].scale_; in AvgPoolingInt8() local
39 double real_multiplier = input_scale / output_scale; in AvgPoolingInt8()
294 float output_scale = pooling_param->quant_args_[1][0].scale_; in MaxPoolingInt8() local
296 double real_multiplier = input_scale / output_scale; in MaxPoolingInt8()
340 float output_scale = pooling_param->quant_args_[1][0].scale_; in MaxPoolingWithQuantInt8() local
342 double real_multiplier = input_scale / output_scale; in MaxPoolingWithQuantInt8()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dsigmoid_int8.cc33 …TableList(int8_t *table, const float input_scale, const int32_t input_zp, const float output_scale, in CalculateTableList() argument
41 const int32_t quantized = std::round(sigmoid_value / output_scale) + output_zp; in CalculateTableList()
53 const float output_scale = output->quant_params().front().scale; in Init() local
55 CalculateTableList(table_list_, input_scale, input_zp, output_scale, output_zp); in Init()
Dhswish_int8.cc39 quant_arg_.output_scale = output->quant_params().front().scale; in Init()
42 …const float output_multiplier = (1.0f / 128.0f) * quant_arg_.input_scale / quant_arg_.output_scale; in Init()
Dscale_int8.cc180 auto output_scale = output->quant_params().front().scale; in InitQuantArgs() local
186 const double input_output_multiplier = input_scale * scale_scale / output_scale; in InitQuantArgs()
198 const double offset_multiplier = offset_scale / output_scale; in InitQuantArgs()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dsync_batch_norm_impl.cu163 __global__ void SyncBatchNormPostBiasScale(size_t C, S *scale, S *bias, S *output_scale, S *output_… in SyncBatchNormPostBiasScale() argument
166 output_scale[pos] = scale[pos]; in SyncBatchNormPostBiasScale()
193 … float *invstds_local, S *scale, S *bias, S *output_scale, S *output_bias, float epsilon, in CalSyncBatchNormPost() argument
198 C, scale, bias, output_scale, output_bias); in CalSyncBatchNormPost()
235 … float *bias, float *output_scale, float *output_bias, float epsilon,
239 … half *bias, half *output_scale, half *output_bias, float epsilon,
243 … float *output_scale, float *output_bias, float epsilon,
247 half *output_scale, half *output_bias, float epsilon,
Dsync_batch_norm_impl.cuh31 … float *invstds_local, S *scale, S *bias, S *output_scale, S *output_bias, float epsilon,
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dbn_fp16_grad.cc96 auto *output_scale = out_tensors_.at(1); in DoExecute() local
111 float16_t *dscale = reinterpret_cast<float16_t *>(output_scale->data()); in DoExecute()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dbn_grad.cc96 auto *output_scale = out_tensors_.at(1); in Execute() local
116 float *dscale = reinterpret_cast<float *>(output_scale->MutableData()); in Execute()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dsync_batch_norm_gpu_kernel.h58 S *output_scale = GetDeviceAddress<S>(outputs, 1); in Launch() local
75 …hNormPost(N_, C_, H_, W_, x, y, means_local, invstds_local, scale, bias, output_scale, output_bias, in Launch()