Searched refs:input_oz_offset (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.c | 293 int input_oz_offset = kernel_unit * kernel_unit * channel; in WinogradWeightTransform() local
301 …MatrixMultiplyWinograd(weight_data + i * input_oz_offset, matrix_gt, tmp_data, kernel_unit, kernel… in WinogradWeightTransform()
312 …MatrixMultiplyWinograd(weight_data + i * input_oz_offset, matrix_gt, tmp_data, kernel_unit, kernel… in WinogradWeightTransform()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 253 int input_oz_offset = kernel_unit * kernel_unit * filter_channel; in WinogradWeightTransformFp16() local
261 …MatrixMultiplyWinogradFp16(weight_data + i * input_oz_offset, matrix_gt_data_fp16, tmp_data, kerne… in WinogradWeightTransformFp16()
272 …MatrixMultiplyWinogradFp16(weight_data + i * input_oz_offset, matrix_gt_data_fp16, tmp_data, kerne… in WinogradWeightTransformFp16()
|