Searched defs:output_min (Results 1 – 5 of 5) 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()
35 __global__ void UpdateInputMinMaxPerLayer(float *output_min, float *output_max, const float min, co… in UpdateInputMinMaxPerLayer()
41 …id UpdateInputMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min, in UpdateInputMinMaxPerChannel()
60 void CalMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min, float… in CalMinMaxPerChannel()
69 void CalMinMaxPerLayer(float *input, float *input_min, float *input_max, float *output_min, float *… in CalMinMaxPerLayer()
|
| /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
|
| D | minmax_update_perlayer_gpu_kernel.cc | 83 float *output_min = GetDeviceAddress<float>(outputs, 0); in Launch() local
|
| /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
233 int8x8_t output_min = vdup_n_s8(out_min); in AvgPoolingOptInt8() local
|
| D | conv3x3_int8.c | 516 int32x4_t output_min = vdupq_n_s32(out_min); in Conv3x3Int8OutputUnit() local
|