Home
last modified time | relevance | path

Searched defs:kernel_w (Results 1 – 25 of 34) sorted by relevance

12

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_slidewindow_fp32.cc45 int kernel_w = filter_tensor->Width(); in Init() local
193 int kernel_w = filter_tensor->Width(); in PackWeight() local
206 int kernel_w = filter_tensor->Width(); in MallocWeightBiasData() local
Dadder_fp32.cc70 int kernel_w = filter_tensor->Width(); in InitWeightBias() local
Ddeconvolution_fp32.cc86 auto kernel_w = weight_tensor->Width(); in PackWeight() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Ddepthwise_conv2d_tests.cc26 OpParameter *CreateParameter(int kernel_h, int kernel_w, int stride_h, int stride_w, int pad_u, int… in CreateParameter()
49 int kernel_w = 3; in TEST_F() local
95 int kernel_w = 3; in TEST_F() local
142 int kernel_w = 2; in TEST_F() local
186 int kernel_w = 3; in TEST_F() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc61 auto kernel_w = weight_tensor->Width(); in PackWeight() local
73 auto kernel_w = weight_tensor->Width(); in MallocWeightBiasData() local
195 auto kernel_w = weight_tensor->Width(); in Init() local
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc59int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetOutputShape()
92 …ing(int64_t in_d, int64_t in_h, int64_t in_w, int64_t kernel_d, int64_t kernel_h, int64_t kernel_w, in GetPadsByPadding()
139 auto kernel_w = kernel_size[2]; in InferShape() local
Davg_pool.cc108 auto kernel_w = kernel_size[3]; in InferShape() local
Dmax_pool.cc103 auto kernel_w = kernel_size[3]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.c352 int kernel_w = conv_param->kernel_w_; in ConvSWFp32() local
434size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv3x32Kernel()
593size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x32Kernel()
688size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x24Kernel()
863size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x24Kernel()
950size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv6x16Kernel()
1128size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x16Kernel()
1208size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv12x8Kernel()
1388size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x8Kernel()
1485size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x8Kernel()
[all …]
Dcommon_func_fp32.h33 size_t kernel_w; member
Dconv_depthwise_fp32.c263 … int kernel_h, int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, in ConvDwCenter()
930int kernel_w, int out_h_step, int block_channel, int in_sh_step, int in_sw_step, int in_kh_step, in DeconvDwCenter()
1067 int kernel_w = conv_param->kernel_w_; in DepthwiseSWAvxFp32() local
1138size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSWWxKKernel()
1199size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW3x32Kernel()
1344size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x32Kernel()
1423size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x24Kernel()
1568size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x24Kernel()
1640size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW4x16Kernel()
1754size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t oc_algin, in DepthwiseSW1x16Kernel()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c33 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp32() local
83 const int kernel_w = conv_param->kernel_w_; in rolling_im2col_hwc() local
154 const int kernel_w = conv_param->kernel_w_; in rolling_im2row_hwc() local
202 const int kernel_w = conv_param->kernel_w_; in col2im_hwc() local
250 const int kernel_w = conv_param->kernel_w_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc90 int32_t kernel_w = filter_tensor_->Width(); in InitWeightBias() local
264 int kernel_w = conv_param->kernel_w_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c32 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp16() local
82 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackUnitFp16() local
149 const int kernel_w = conv_param->kernel_w_; in RollingCol2ImPackUnitFp16() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc64 int kernel_w = conv_param->kernel_w_; in TEST_F() local
127 int kernel_w = conv_param->kernel_w_; in TEST_F() local
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv_init_int8_wrapper.c23int kernel_w, int input_channel, int output_channel, int32_t input_zp, bool filter_peroc, in ConvInit()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc82 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
Dconvolution_fp32_coder.cc60 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
Dconv2d_delegate_fp32_coder.cc78 int kernel_w = conv_param->kernel_w_; in CPUConvolutionFP32CoderSelect() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c54 int kernel_w = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(weight); in Deconv2dInferShape() local
Dconv2d_infer.c20 int kernel_w = param->kernel_w_; in ConvInferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c47 int kernel_w = conv_param->kernel_w_; in SelectOutputUnit() local
/third_party/mindspore/mindspore/core/ops/fusion/
Dmax_pool_fusion.cc74 auto kernel_w = kernel_size[3]; in InferShape() local
Davg_pool_fusion.cc77 auto kernel_w = kernel_size[3]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/intrinsics/sse/
DDepthwiseFp32_Sse.c111 … size_t kernel_h, size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, in ConvDwFp32Center()
233size_t kernel_w, size_t out_h_step, size_t block_channel, size_t in_sh_step, size_t in_sw_step, in DeconvDwFp32Center()

12