Searched refs:trans_out_data1 (Results 1 – 2 of 2) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
D | minimal_filtering_generator.c | 284 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/ |
D | winograd_transform_fp16.c | 243 …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()
|