Searched defs:input_max (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | minmax_update_impl.cu | 24 __global__ void UpdateInputMinMaxPerLayerWithEMA(const float *input_min, const float *input_max, fl… in UpdateInputMinMaxPerLayerWithEMA()
40 __global__ void UpdateInputMinMaxPerChannel(float *input, float *input_min, float *input_max, float… in UpdateInputMinMaxPerChannel()
59 cudaError_t CalMinMaxPerChannel(float *input, float *input_min, float *input_max, float *output_min… in CalMinMaxPerChannel()
68 cudaError_t CalMinMaxPerLayer(float *input, float *input_min, float *input_max, float *output_min, … in CalMinMaxPerLayer()
|
| D | fake_quant_perlayer_impl.cu | 57 __global__ void NudgeMinMaxPerLayer(float *input_min, float *input_max, const float quant_min, cons… in NudgeMinMaxPerLayer()
107 cudaError_t CalNudgePerLayer(float *input_min, float *input_max, const float quant_min, const float… in CalNudgePerLayer()
|
| D | fake_quant_perchannel_impl.cu | 36 __global__ void NudgeMinMaxPerChannel(float *input_min, float *input_max, const float quant_min, co… in NudgeMinMaxPerChannel()
70 cudaError_t CalNudgePerChannel(float *input_min, float *input_max, const float quant_min, const flo… in CalNudgePerChannel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/quant/ |
| D | fake_quant_perchannel_gpu_kernel.cc | 96 … float *input_max, float *nudge_min, float *nudge_max, in CalFakeQuantize()
116 float *input_max = GetDeviceAddress<float>(inputs, kIndex2); in Launch() local
|
| D | minmax_update_perlayer_gpu_kernel.cc | 67 float *input_max = GetDeviceAddress<float>(inputs, kIndex2); in Launch() local
|
| D | minmax_update_perchannel_gpu_kernel.cc | 71 float *input_max = GetDeviceAddress<float>(inputs, kIndex2); in Launch() local
|
| D | fake_quant_perlayer_gpu_kernel.cc | 100 float *input_max = GetDeviceAddress<float>(inputs, kIndex2); in Launch() local
|
| D | fake_quant_perchannel_grad_gpu_kernel.cc | 98 float *input_max = GetDeviceAddress<float>(inputs, kIndex3); in Launch() local
|
| D | fake_quant_perlayer_grad_gpu_kernel.cc | 101 float *input_max = GetDeviceAddress<float>(inputs, kIndex3); in Launch() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_parameterized_truncated_normal.py | 27 … def __init__(self, input_shape, input_mean, input_stdevs, input_min, input_max, seed=0, seed2=0): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | random_uniform_int.cc | 40 auto input_max = ctx.Input(kMaxIdx); in Compute() local
|