Searched refs:per_channel_size (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | prelu_impl.cu | 20 __global__ void CalPReLUKernel(size_t size, size_t weight_size, size_t per_channel_size, in CalPReLUKernel() argument
23 size_t channel_id = weight_size == 1 ? 0 : (pos / per_channel_size) % weight_size; in CalPReLUKernel()
29 void CalPReLU(size_t size, size_t weight_size, size_t per_channel_size, in CalPReLU() argument
31 …PReLUKernel<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, weight_size, per_channel_size, in CalPReLU()
|
| D | prelu_grad_impl.cu | 22 __global__ void CalPReLUGradKernel(size_t size, size_t weight_size, size_t per_channel_size, in CalPReLUGradKernel() argument
26 size_t channel_id = weight_size == 1 ? 0 : (pos / per_channel_size) % weight_size; in CalPReLUGradKernel()
54 void CalPReLUGrad(size_t size, size_t weight_size, size_t per_channel_size, in CalPReLUGrad() argument
59 …UGradKernel<<<GET_BLOCKS(size), GET_THREADS, 0, cuda_stream>>>(size, weight_size, per_channel_size, in CalPReLUGrad()
|
| D | prelu_impl.cuh | 23 void CalPReLU(size_t input_size, size_t weight_size, size_t per_channel_size,
|
| D | prelu_grad_impl.cuh | 23 void CalPReLUGrad(size_t input_size, size_t weight_size, size_t per_channel_size,
|
| /third_party/mindspore/mindspore/lite/micro/coder/wrapper/fp32/ |
| D | dequant_int8_to_fp32_wrapper.c | 41 size_t per_channel_size = de_quant_nums / channels; in DequantData() local
51 for (size_t j = 0; j < per_channel_size; j++) { in DequantData()
52 float dequant_data = (quant_src[per_channel_size * i + j] - zero_point) * scale; in DequantData()
53 de_quant_dst[per_channel_size * i + j] = dequant_data * var_corr + mean_corr; in DequantData()
|
| /third_party/mindspore/mindspore/lite/src/ |
| D | weight_decoder.h | 166 size_t per_channel_size = input_tensor->ElementsNum() / channels; variable
177 for (size_t j = 0; j < per_channel_size; j++) {
178 auto index = per_channel_size * i + j;
|