Searched refs:trans_out_data (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.c | 271 float *trans_out_data = (float *)(malloc(channel * input_unit * input_unit * sizeof(float))); in WinogradWeightTransform() local
272 if (trans_out_data == NULL) { in WinogradWeightTransform()
281 free(trans_out_data); in WinogradWeightTransform()
288 free(trans_out_data); in WinogradWeightTransform()
309 PackHWCToWHC(trans_out_data1, trans_out_data, input_unit, input_unit, channel); in WinogradWeightTransform()
315 …MatrixMultiplyWinograd(tmp_data, matrix_gt, trans_out_data, input_unit, kernel_unit, input_unit, c… in WinogradWeightTransform()
323 *(winograd_data + output_oz_offset + c * oc_block) = trans_out_data[in_offset + c]; in WinogradWeightTransform()
330 memcpy(winograd_data + i * channel * input_unit * input_unit, trans_out_data, in WinogradWeightTransform()
339 free(trans_out_data); in WinogradWeightTransform()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 228 …float16_t *trans_out_data = (float16_t *)(malloc(filter_channel * input_unit * input_unit * sizeof… in WinogradWeightTransformFp16() local
229 if (trans_out_data == NULL) { in WinogradWeightTransformFp16()
240 free(trans_out_data); in WinogradWeightTransformFp16()
248 free(trans_out_data); in WinogradWeightTransformFp16()
269 PackHWCToWHCFp16(trans_out_data1, trans_out_data, input_unit, input_unit, filter_channel); in WinogradWeightTransformFp16()
275 …MatrixMultiplyWinogradFp16(tmp_data, matrix_gt_data_fp16, trans_out_data, input_unit, kernel_unit,… in WinogradWeightTransformFp16()
284 *(winograd_data + output_oz_offset + c * oc_block) = trans_out_data[in_offset + c]; in WinogradWeightTransformFp16()
291 memcpy(winograd_data + i * filter_channel * input_unit * input_unit, trans_out_data, in WinogradWeightTransformFp16()
301 free(trans_out_data); in WinogradWeightTransformFp16()
|