Searched refs:data_col (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pack_ext.c | 49 float *data_col = data_col_orig + i * channels * stride; in RollingIm2ColPackDwUnitFp32() local
57 data_col[c * stride] = in_data[offset + c]; in RollingIm2ColPackDwUnitFp32()
59 data_col++; in RollingIm2ColPackDwUnitFp32()
62 data_col[c * stride] = 0; in RollingIm2ColPackDwUnitFp32()
64 data_col++; in RollingIm2ColPackDwUnitFp32()
71 void rolling_im2col_hwc(const float *in_data, float *data_col, const ConvParameter *conv_param, int… in rolling_im2col_hwc() argument
106 *data_col = in_data[offset]; in rolling_im2col_hwc()
107 data_col++; in rolling_im2col_hwc()
109 *data_col = 0; in rolling_im2col_hwc()
110 data_col++; in rolling_im2col_hwc()
[all …]
|
| D | pack_ext.h | 32 void rolling_im2col_hwc(const float *in_data, float *data_col, const ConvParameter *conv_param, int…
34 void rolling_col2im_hwc(const float *data_col, float *data_im, const ConvParameter *conv_param, int…
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pack_fp16_ext.c | 48 float16_t *data_col = data_col_orig + i * channels * stride; in RollingIm2ColPackDwUnitFp16() local
56 data_col[c * stride] = in_data[offset + c]; in RollingIm2ColPackDwUnitFp16()
58 data_col++; in RollingIm2ColPackDwUnitFp16()
61 data_col[c * stride] = 0; in RollingIm2ColPackDwUnitFp16()
63 data_col++; in RollingIm2ColPackDwUnitFp16()
137 void RollingCol2ImPackUnitFp16(const float16_t *data_col, float16_t *data_im, const ConvParameter *… in RollingCol2ImPackUnitFp16() argument
173 *data_im_ptr += *data_col; in RollingCol2ImPackUnitFp16()
175 data_col++; in RollingCol2ImPackUnitFp16()
193 data_im_ptr[i] += data_col[i]; in RollingCol2ImPackUnitFp16()
196 data_col += channels; in RollingCol2ImPackUnitFp16()
|
| D | pack_fp16_ext.h | 31 void RollingCol2ImPackUnitFp16(const float16_t *data_col, float16_t *data_im, const ConvParameter *…
|
| /third_party/boost/libs/multi_array/test/ |
| D | storage_order.cpp | 38 int data_col[] = { in main() local
64 A.assign(data_col,data_col+num_elements); in main()
85 A.assign(data_col,data_col+num_elements); in main()
|
| /third_party/boost/boost/graph/parallel/ |
| D | distribution.hpp | 368 std::size_t data_col = i % data_columns_per_row; in local() local
370 + (data_col % block_columns); in local()
416 std::size_t data_col = i % data_columns_per_row; in get_block_num() local
418 std::size_t block_col = data_col / block_columns; in get_block_num()
514 std::size_t data_col = i % data_columns_per_row; in local() local
516 + (data_col % block_columns); in local()
527 std::size_t data_col = i % data_columns_per_row; in get_block_num() local
529 std::size_t block_col = data_col / block_columns; in get_block_num()
|
| /third_party/mindspore/mindspore/dataset/engine/ |
| D | datasets.py | 6161 data_col = input_data[data_keys[0]]
6162 if hasattr(data_col, "values"):
|