Searched refs:output_min (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | minmax_update_impl.cu | 25 …UpdateInputMinMaxPerLayerWithEMA(const float *input_min, const float *input_max, float *output_min, in UpdateInputMinMaxPerLayerWithEMA() argument
28 output_min[0] = decay * (min) + (1 - decay) * (input_min[0]); in UpdateInputMinMaxPerLayerWithEMA()
29 output_min[0] = output_min[0] > 0 ? 0 : output_min[0]; in UpdateInputMinMaxPerLayerWithEMA()
35 __global__ void UpdateInputMinMaxPerLayer(float *output_min, float *output_max, const float min, co… in UpdateInputMinMaxPerLayer() argument
36 output_min[0] = min > 0 ? 0 : min; in UpdateInputMinMaxPerLayer()
41 …id UpdateInputMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min, in UpdateInputMinMaxPerChannel() argument
48 output_min[i] = ema_decay * sum.first[0] + (1 - ema_decay) * input_min[i]; in UpdateInputMinMaxPerChannel()
51 output_min[i] = sum.first[0]; in UpdateInputMinMaxPerChannel()
54 output_min[i] = output_min[i] > 0 ? 0 : output_min[i]; in UpdateInputMinMaxPerChannel()
60 void CalMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min, float… in CalMinMaxPerChannel() argument
[all …]
|
| D | minmax_update_impl.cuh | 22 void CalMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min, float…
26 void CalMinMaxPerLayer(float *input, float *input_min, float *input_max, float *output_min, float *…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | minmax_update_perchannel_gpu_kernel.cc | 89 float *output_min = GetDeviceAddress<float>(outputs, 0); in Launch() local
96 …CalMinMaxPerChannel(input, input_min, input_max, output_min, output_max, input_size_ / sizeof(floa… in Launch()
|
| D | minmax_update_perlayer_gpu_kernel.cc | 83 float *output_min = GetDeviceAddress<float>(outputs, 0); in Launch() local
89 …CalMinMaxPerLayer(input, input_min, input_max, output_min, output_max, quant_num_, ema_decay_, ema… in Launch()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 173 int8x8_t output_min = vdup_n_s8(out_min); in AvgPoolingOptInt8() local
177 real_out[0] = vmax_s8(real_out[0], output_min); in AvgPoolingOptInt8()
181 real_out[1] = vmax_s8(real_out[1], output_min); in AvgPoolingOptInt8()
233 int8x8_t output_min = vdup_n_s8(out_min); in AvgPoolingOptInt8() local
237 real_out = vmax_s8(real_out, output_min); in AvgPoolingOptInt8()
|
| D | conv3x3_int8.c | 516 int32x4_t output_min = vdupq_n_s32(out_min); in Conv3x3Int8OutputUnit() local
526 d00 = vmaxq_s32(d00, output_min); in Conv3x3Int8OutputUnit()
536 d01 = vmaxq_s32(d01, output_min); in Conv3x3Int8OutputUnit()
546 d10 = vmaxq_s32(d10, output_min); in Conv3x3Int8OutputUnit()
556 d11 = vmaxq_s32(d11, output_min); in Conv3x3Int8OutputUnit()
|