Searched defs:data_col (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/ |
| D | pack_ext.c | 48 float *data_col = data_col_orig + i * channels * stride; in RollingIm2ColPackDwUnitFp32() local
70 void rolling_im2col_hwc(const float *in_data, float *data_col, const ConvParameter *conv_param, int… in rolling_im2col_hwc()
190 void col2im_hwc(const float *data_col, float *data_im, const ConvParameter *conv_param) { in col2im_hwc()
238 void rolling_col2im_hwc(const float *data_col, float *data_im, const ConvParameter *conv_param, int… in rolling_col2im_hwc()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/ |
| D | pack_fp16_ext.c | 48 float16_t *data_col = data_col_orig + i * channels * stride; in RollingIm2ColPackDwUnitFp16() local
137 void RollingCol2ImPackUnitFp16(const float16_t *data_col, float16_t *data_im, const ConvParameter *… in RollingCol2ImPackUnitFp16()
|