Home
last modified time | relevance | path

Searched refs:C3NUM (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/avx/
DDeconvMatMulAvx.c33 __m256 tmp3 = _mm256_set1_ps(*(src + C3NUM)); in Deconv4X8AvxKernel()
64 __m256 tmp3 = _mm256_set1_ps(*(src + C3NUM)); in Deconv4X16AvxKernel()
117 tmp = _mm256_set1_ps(*(src + C3NUM)); in Deconv4X24AvxKernel()
147 col_num = MSMIN(C3NUM, col_block - col_tmp); in DeconvMatmulAvx()
162 __m256 w[C3NUM]; in DeconvColXRowAvxKernel()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Ddeconvolution_fp32.cc113 …hread_stride_ = UP_DIV(UP_DIV(conv_param_->output_channel_, C8NUM * C3NUM), thread_count_) * C3NUM; in InitParam()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/
Dop_base.h31 #define C3NUM 3 macro
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.c297 } else if (real_cal_row == C3NUM) { in CommonConv6x16Kernel()
440 float *dst_4 = dst + out_step * C3NUM; in SWConv3x32Kernel()
599 float *dst_4 = dst + out_step * C3NUM; in SWConv1x32Kernel()
Dpack_fp32.c436 oc_block = MSMIN(C3NUM, oc_block_num - i); in PackNHWCToCXHWNXFp32()
485 oc_block = MSMIN(C3NUM, oc_block_num - i); // max_tile = 4 in PackNHWCToCXHWNXFp32()