Home
last modified time | relevance | path

Searched refs:trans_out_data1 (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dminimal_filtering_generator.c284 float *trans_out_data1 = (float *)(malloc(channel * input_unit * input_unit * sizeof(float))); in WinogradWeightTransform() local
285 if (trans_out_data1 == NULL) { in WinogradWeightTransform()
306 …MatrixMultiplyWinograd(tmp_data1, matrix_gt, trans_out_data1, input_unit, kernel_unit, input_unit,… in WinogradWeightTransform()
309 PackHWCToWHC(trans_out_data1, trans_out_data, input_unit, input_unit, channel); in WinogradWeightTransform()
336 free(trans_out_data1); in WinogradWeightTransform()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/
Dwinograd_transform_fp16.c243 …float16_t *trans_out_data1 = (float16_t *)(malloc(filter_channel * input_unit * input_unit * sizeo… in WinogradWeightTransformFp16() local
244 if (trans_out_data1 == NULL) { in WinogradWeightTransformFp16()
266 …MatrixMultiplyWinogradFp16(tmp_data1, matrix_gt_data_fp16, trans_out_data1, input_unit, kernel_uni… in WinogradWeightTransformFp16()
269 PackHWCToWHCFp16(trans_out_data1, trans_out_data, input_unit, input_unit, filter_channel); in WinogradWeightTransformFp16()
298 free(trans_out_data1); in WinogradWeightTransformFp16()