Home
last modified time | relevance | path

Searched refs:kernel_unit (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.c208 … const float *matrix_gt, int oc_block, int input_unit, int kernel_unit, in WinogradWeightTransformFp16() argument
215 …float16_t *matrix_gt_data_fp16 = (float16_t *)(malloc(input_unit * kernel_unit * sizeof(float16_t)… in WinogradWeightTransformFp16()
219 Float32ToFloat16(matrix_gt, matrix_gt_data_fp16, input_unit * kernel_unit); in WinogradWeightTransformFp16()
223 …float16_t *tmp_data = (float16_t *)(malloc(filter_channel * input_unit * kernel_unit * sizeof(floa… in WinogradWeightTransformFp16()
236 …float16_t *tmp_data1 = (float16_t *)(malloc(filter_channel * input_unit * kernel_unit * sizeof(flo… in WinogradWeightTransformFp16()
253 int input_oz_offset = kernel_unit * kernel_unit * filter_channel; in WinogradWeightTransformFp16()
261 …MultiplyWinogradFp16(weight_data + i * input_oz_offset, matrix_gt_data_fp16, tmp_data, kernel_unit, in WinogradWeightTransformFp16()
262 kernel_unit, input_unit, filter_channel); in WinogradWeightTransformFp16()
264 PackHWCToWHCFp16(tmp_data, tmp_data1, kernel_unit, input_unit, filter_channel); in WinogradWeightTransformFp16()
266 …yWinogradFp16(tmp_data1, matrix_gt_data_fp16, trans_out_data1, input_unit, kernel_unit, input_unit, in WinogradWeightTransformFp16()
[all …]
Dwinograd_transform_fp16.h46 … const float *matrix_gt, int oc_block, int input_unit, int kernel_unit,
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dminimal_filtering_generator.c256 … int oc_block, int input_unit, int kernel_unit, int channel, int batch, bool pack) { in WinogradWeightTransform() argument
267 float *tmp_data = (float *)(malloc(channel * input_unit * kernel_unit * sizeof(float))); in WinogradWeightTransform()
278 float *tmp_data1 = (float *)(malloc(channel * input_unit * kernel_unit * sizeof(float))); in WinogradWeightTransform()
293 int input_oz_offset = kernel_unit * kernel_unit * channel; in WinogradWeightTransform()
301 …inograd(weight_data + i * input_oz_offset, matrix_gt, tmp_data, kernel_unit, kernel_unit, input_un… in WinogradWeightTransform()
304 PackHWCToWHC(tmp_data, tmp_data1, kernel_unit, input_unit, channel); in WinogradWeightTransform()
306 …MatrixMultiplyWinograd(tmp_data1, matrix_gt, trans_out_data1, input_unit, kernel_unit, input_unit,… in WinogradWeightTransform()
312 …inograd(weight_data + i * input_oz_offset, matrix_gt, tmp_data, kernel_unit, kernel_unit, input_un… in WinogradWeightTransform()
315 …MatrixMultiplyWinograd(tmp_data, matrix_gt, trans_out_data, input_unit, kernel_unit, input_unit, c… in WinogradWeightTransform()