Home
last modified time | relevance | path

Searched defs:matrix_gt (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Dconvolution_winograd_fp16.cc24const float *matrix_gt, int oc_block) { in WinogradFilterTransformFp16()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_winograd_base_fp32.cc36const float *matrix_gt, int oc_block) { in WinogradFilterTransform()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dminimal_filtering_generator.c182 float *matrix_gt, float coefficient, int out_unit, int filter_size) { in CookToomFilter()
256 …tTransform(const float *weight_data, float *winograd_data, float *matrix_g, const float *matrix_gt, in WinogradWeightTransform()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/
Dwinograd_transform_fp16.c264const float *matrix_gt, int oc_block, int input_unit, int kernel_unit, in WinogradWeightTransformFp16()
Ddeconv_winograd_fp16.c268 float matrix_gt[64]; in PackDeConvWgDataFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Ddeconv_winograd_fp32.c52 float matrix_gt[64], matrix_at[64], matrix_bt[64]; in PackDeConvWgDataFp32() local