Home
last modified time | relevance | path

Searched defs:output_scale (Results 1 – 18 of 18) sorted by relevance

/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()
49 const float output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
/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()
53 const float output_scale = output->quant_params().front().scale; in Init() local
Dscale_int8.cc180 auto output_scale = output->quant_params().front().scale; in InitQuantArgs() local
/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
Dleaky_relu_int8.c24 float output_scale = quant_prelu_parm->quant_arg.out_args_.scale_; in DoLeakReluInt8() local
Dsqueeze_int8.c21 float output_scale = quant_Squeeze_parm->out_quant_args_->scale_; in SqueezeInt8() local
Dhswish_int8.h27 double output_scale; member
Dpower_int8.c22 double output_scale = param->quant_arg_.out_args_.scale_; in PowerInt8() local
Dconcat_int8.c23 float output_scale = para->quant_arg_.out_args_.scale_; in Int8Concat() local
Dslice_int8.c23 double output_scale = param->quant_arg_.out_args_.scale_; in SliceInt8() local
Dpooling_int8.c37 float output_scale = pooling_param->quant_args_[1][0].scale_; in AvgPoolingInt8() local
294 float output_scale = pooling_param->quant_args_[1][0].scale_; in MaxPoolingInt8() local
340 float output_scale = pooling_param->quant_args_[1][0].scale_; in MaxPoolingWithQuantInt8() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/cmsis-nn/int8/
Dmul_int8_coder.cc42 const double output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
Dadd_int8_coder.cc46 const double output_scale = output_tensor_->quant_params().at(0).scale; in Prepare() local
Dconv2d_base_coder.cc31 auto output_scale = static_cast<double>(conv_quant_arg_->output_quant_args_[0].scale_); in SetQuantArgs() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dbn_grad.cc96 auto *output_scale = out_tensors_.at(1); in Execute() local
/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
/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()
193 … float *invstds_local, S *scale, S *bias, S *output_scale, S *output_bias, float epsilon, in CalSyncBatchNormPost()
/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