Searched defs:oc_block_num (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_slidewindow_fp32.cc | 47 int oc_block_num = UP_DIV(output_channel, oc_tile_); in Init() local
142 int oc_block_num = UP_DIV(out_channel, oc_tile_); in InitTmpBuffer() local
194 int oc_block_num = UP_DIV(output_channel, oc_tile_); in PackWeight() local
210 int oc_block_num = UP_DIV(output_channel, oc_tile_); in MallocWeightBiasData() local
|
| D | convolution_fp32.cc | 85 size_t oc_block_num = UP_ROUND(out_channel, OC_BLOCK); in Init() local
207 size_t oc_block_num = UP_ROUND(out_channel, OC_BLOCK); in MallocWeightBiasData() local
|
| D | adder_fp32.cc | 78 int oc_block_num = UP_DIV(out_channel, C4NUM); in InitWeightBias() local
|
| D | matmul_fp32_base.cc | 421 int oc_block_num = UP_DIV(out_channel, col_tile_); in InitTmpOutBuffer() local
|
| D | lstm_fp32.cc | 311 int oc_block_num = UP_DIV(out_channel, state_col_tile_); in MallocRunBuffer() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/ |
| D | convolution_winograd_fp16.cc | 41 int oc_block_num = UP_DIV(out_channel, col_tile_); in MallocWeightBiasData() local
155 int oc_block_num = UP_DIV(out_channel, col_tile_); in Init() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/ |
| D | convolution_fp32_coder.cc | 70 int oc_block_num = UP_ROUND(out_channel, oc_block); in InitWeightBias() local
|
| D | convolution_winograd_fp32_coder.cc | 131 int oc_block_num = UP_DIV(out_channel, C8NUM); in InitWeightBias() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | pack_fp32.c | 162 void PackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int inp… in PackNHWCToNXHWCXFp32()
234 void SWPackNHWCToNXHWCXFp32(int kernel_h, int kernel_w, int output_channel, int oc_block_num, int i… in SWPackNHWCToNXHWCXFp32()
433 int oc_block_num = UP_DIV(channel, C8NUM); in PackNHWCToCXHWNXFp32() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | winograd_transform_fp16.c | 211 int oc_block_num = UP_DIV(filter_batch, oc_block); in WinogradWeightTransformFp16() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | minimal_filtering_generator.c | 261 int oc_block_num = UP_DIV(batch, oc_block); in WinogradWeightTransform() local
|