Searched refs:winograd_data (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.h | 45 int WinogradWeightTransformFp16(const float16_t *weight_data, float16_t *winograd_data, const float…
|
| D | winograd_transform_fp16.c | 207 int WinogradWeightTransformFp16(const float16_t *weight_data, float16_t *winograd_data, const float… in WinogradWeightTransformFp16() argument
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()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.c | 255 int WinogradWeightTransform(const float *weight_data, float *winograd_data, float *matrix_g, const … in WinogradWeightTransform() argument
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()
|
| D | minimal_filtering_generator.h | 52 int WinogradWeightTransform(const float *weight_data, float *winograd_data, float *matrix_g, const …
|