Searched defs:oc_algin (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/ |
| D | convolution_depthwise_slidewindow_x86_fp32.cc | 48 int oc_algin = UP_DIV(conv_param_->output_channel_, oc_tile_); in InitPackedInputOutput() local
68 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in Init() local
169 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in PackWeight() local
179 int oc_algin = UP_DIV(weight_tensor->Batch(), oc_tile_); in MallocWeightBiasData() local
|
| D | convolution_fp32.cc | 64 int oc_algin = UP_DIV(conv_param_->output_channel_, OC_BLOCK); in InitTmpBuffer() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/ |
| D | conv_depthwise_fp32.c | 1069 int oc_algin = sw_param->block_channel_; in DepthwiseSWAvxFp32() local
1138 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSWWxKKernel()
1199 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW3x32Kernel()
1344 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x32Kernel()
1423 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x24Kernel()
1568 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x24Kernel()
1640 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x16Kernel()
1754 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x16Kernel()
1819 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW8x8Kernel()
1940 … size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x8Kernel()
|