Searched refs:out_scale_ (Results 1 – 17 of 17) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | exp_fp32.c | 63 if (param->out_scale_ != 1) { in ExpFusionFp32()
66 … MS_FLOAT32X4 scale = {param->out_scale_, param->out_scale_, param->out_scale_, param->out_scale_}; in ExpFusionFp32()
75 dst[i] *= param->out_scale_; in ExpFusionFp32()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | exp_fp32_coder.cc | 30 exp_parameter_->out_scale_ = 1; in Prepare()
33 exp_parameter_->out_scale_ = expf(exp_parameter_->shift_); in Prepare()
35 exp_parameter_->out_scale_ = powf(exp_parameter_->base_, exp_parameter_->shift_); in Prepare()
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | exp_fp32.cc | 33 param_->out_scale_ = 1; in Init()
36 param_->out_scale_ = expf(param_->shift_); in Init()
38 param_->out_scale_ = powf(param_->base_, param_->shift_); in Init()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | exp_fp16.c | 65 if (param->out_scale_ != 1) { in ExpFusionFp16()
68 MS_FLOAT16X8 scale = MS_MOVQ_F16(param->out_scale_); in ExpFusionFp16()
77 dst[i] *= param->out_scale_; in ExpFusionFp16()
|
| /third_party/mindspore/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/operator_library/kernels/nnacl/fp32/ |
| D | exp_fp32.c | 32 if (parameter->out_scale_ != 1) { in Exp()
34 output_data[i] = output_data[i] * parameter->out_scale_; in Exp()
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | reduce_int8_coder.cc | 31 quant_arg_.out_scale_ = output_quant.scale; in CalculateQuantArgs()
33 const double input_output_multiplier = quant_arg_.in_scale_ / quant_arg_.out_scale_; in CalculateQuantArgs()
89 …uble sum_square_multiplier = quant_arg_.in_scale_ * (quant_arg_.in_scale_ / quant_arg_.out_scale_); in CalculateQuantArgs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/ |
| D | exp_parameter.h | 29 float out_scale_; member
|
| D | layer_norm_parameter.h | 46 double out_scale_; member
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | reduce_int8.cc | 213 bias_ = quant_arg_.out_zp_ - quant_arg_.in_zp_ * quant_arg_.in_scale_ / quant_arg_.out_scale_; in ReduceMean4DCalQuantParam()
215 double reciprocal = quant_arg_.in_scale_ / (quant_arg_.out_scale_ * reduce_num); in ReduceMean4DCalQuantParam()
235 quant_arg_.out_scale_ = output->quant_params().front().scale; in CalculateQuantArgs()
239 const double input_output_multiplier = quant_arg_.in_scale_ / quant_arg_.out_scale_; in CalculateQuantArgs()
317 …le last_sumsquare_multiplier = quant_arg_.in_scale_ * quant_arg_.in_scale_ / quant_arg_.out_scale_; in CalculateQuantArgsReduceSumSquare()
|
| D | tanh_int8.cc | 29 tanh_quant_.out_scale_ = output->quant_params().front().scale; in Init()
|
| D | layer_norm_int8.cc | 63 quant_param_->out_scale_ = output->quant_params().front().scale; in SetQuantArgs()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | tanh_int8.h | 30 double out_scale_; member
|
| D | tanh_int8.c | 26 int32_t int32_dst = (int32_t)round(fp32_dst * 1.0 / quant->out_scale_ + quant->out_zp_); in TanhInt8()
|
| D | layer_norm_int8.c | 25 int32_t int32_dst = (int32_t)round(fp32_dst * 1.0 / quant->out_scale_ + quant->out_zp_); in LayerNormGammaAndBetaInt8()
|
| D | quantize.h | 141 double out_scale_; member
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/serializers/nnacl_serializer/ |
| D | nnacl_fp32_serializer.cc | 133 exp_parameter.out_scale_, exp_parameter.element_num_); in CodeStruct()
|
| D | nnacl_int8_serializer.cc | 190 …eQuantArg", name, reduce_quant_arg.in_scale_, reduce_quant_arg.in_zp_, reduce_quant_arg.out_scale_, in CodeStruct()
|