Searched defs:output_max (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | minmax_update_impl.cu | 26 … float *output_max, const float min, const float max, in UpdateInputMinMaxPerLayerWithEMA()
35 __global__ void UpdateInputMinMaxPerLayer(float *output_min, float *output_max, const float min, co… in UpdateInputMinMaxPerLayer()
42 … float *output_max, int channels, int per_channel_nums, bool ema, in UpdateInputMinMaxPerChannel()
60 …xPerChannel(float *input, float *input_min, float *input_max, float *output_min, float *output_max, in CalMinMaxPerChannel()
69 …MaxPerLayer(float *input, float *input_min, float *input_max, float *output_min, float *output_max, in CalMinMaxPerLayer()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/ |
| D | minmax_update_perchannel_gpu_kernel.cc | 90 float *output_max = GetDeviceAddress<float>(outputs, 1); in Launch() local
|
| D | minmax_update_perlayer_gpu_kernel.cc | 84 float *output_max = GetDeviceAddress<float>(outputs, 1); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | pooling_int8.c | 174 int8x8_t output_max = vdup_n_s8(out_max); in AvgPoolingOptInt8() local
234 int8x8_t output_max = vdup_n_s8(out_max); in AvgPoolingOptInt8() local
|
| D | conv3x3_int8.c | 517 int32x4_t output_max = vdupq_n_s32(out_max); in Conv3x3Int8OutputUnit() local
|