| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/ |
| D | convolution_slidewindow_fp32.cc | 52 int oc_block_num = UP_DIV(output_channel, oc_tile_); in Prepare() local
148 int oc_block_num = UP_DIV(out_channel, oc_tile_); in InitTmpBuffer() local
200 int oc_block_num = UP_DIV(output_channel, oc_tile_); in PackWeight() local
217 int oc_block_num = UP_DIV(output_channel, oc_tile_); in MallocWeightBiasData() local
|
| D | convolution_im2col_base_fp32.cc | 86 size_t oc_block_num = UP_ROUND(out_channel, oc_tile_); in Prepare() local
220 size_t oc_block_num = UP_ROUND(out_channel, oc_tile_); in MallocWeightBiasData() local
|
| D | convolution_fp32.cc | 101 size_t oc_block_num = UP_ROUND(out_channel, OC_BLOCK); in Prepare() local
274 size_t oc_block_num = UP_ROUND(out_channel, OC_BLOCK); in MallocWeightBiasData() local
|
| D | adder_fp32.cc | 79 int oc_block_num = UP_DIV(out_channel, C4NUM); in InitWeightBias() local
|
| D | lstm_fp32_base.cc | 223 int oc_block_num = UP_DIV(out_channel, state_col_tile_); in MallocRunBuffer() local
|
| D | matmul_fp32_base.cc | 743 int oc_block_num = UP_DIV(out_channel, col_tile_); in InitTmpOutBuffer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/ |
| D | convolution_slidewindow.c | 53 int oc_block_num = UP_DIV(compute->out_c_, conv_sw->oc_tile_); in ConvSWInitTmpBuffer() local
91 int oc_block_num = UP_DIV(output_channel, conv_sw->oc_tile_); in ConvSWPackWeight() local
114 int oc_block_num = UP_DIV(output_channel, conv_sw->oc_tile_); in ConvSWMallocWeightBiasData() local
216 int oc_block_num = UP_DIV(output_channel, conv_sw->oc_tile_); in ConvolutionSWPrepare() local
|
| D | convolution_im2col_base.c | 51 size_t oc_block_num = UP_ROUND(conv->compute_.out_c_, conv_im2col->oc_tile_); in ConvIm2ColBaseMallocWeightBiasData() local
218 int oc_block_num = UP_ROUND(conv_im2col->conv_.compute_.out_c_, conv_im2col->oc_tile_); in ConvolutionIm2colBasePrepare() local
|
| D | matmul_base.c | 456 int oc_block_num = UP_DIV(out_channel, matmul->compute_.col_tile_); in MatmulBaseInitTmpOutBuffer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/ |
| D | convolution_winograd_fp16.cc | 42 int oc_block_num = UP_DIV(out_channel, col_tile_); in MallocWeightBiasData() local
181 int oc_block_num = UP_DIV(out_channel, col_tile_); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/ |
| D | convolution_fp16_coder.cc | 59 int oc_block_num = UP_ROUND(out_channel, C8NUM); in InitWeightBias() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_fp32_coder.cc | 71 int oc_block_num = UP_ROUND(out_channel, oc_block); in InitWeightBias() local
|
| D | convolution_winograd_fp32_coder.cc | 127 int oc_block_num = UP_DIV(out_channel, col_tile_); in InitParameter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | minimal_filtering_generator.c | 262 int oc_block_num = UP_DIV(batch, oc_block); in WinogradWeightTransform() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 267 int oc_block_num = UP_DIV(filter_batch, oc_block); in WinogradWeightTransformFp16() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/ |
| D | pack_fp32.c | 1154 void PackNHWCToNXHWCXFp32H1W1(int output_channel, int oc_block_num, int input_channel, float *tmp_w… in PackNHWCToNXHWCXFp32H1W1()
1190 void PackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int inp… in PackNHWCToNXHWCXFp32()
1250 void SWPackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int i… in SWPackNHWCToNXHWCXFp32()
1516 int oc_block_num = UP_DIV(channel, C8NUM); in PackNHWCToCXHWNXFp32() local
|