Home
last modified time | relevance | path

Searched defs:kernel_h (Results 1 – 25 of 33) sorted by relevance

12

/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dconvolution_slidewindow_fp32.cc44 int kernel_h = filter_tensor->Height(); in Init() local
192 int kernel_h = filter_tensor->Height(); in PackWeight() local
205 int kernel_h = filter_tensor->Height(); in MallocWeightBiasData() local
Dadder_fp32.cc69 int kernel_h = filter_tensor->Height(); in InitWeightBias() local
Ddeconvolution_fp32.cc85 auto kernel_h = weight_tensor->Height(); 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()
48 int kernel_h = 3; in TEST_F() local
94 int kernel_h = 3; in TEST_F() local
141 int kernel_h = 2; in TEST_F() local
185 int kernel_h = 3; in TEST_F() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16/
Ddeconvolution_fp16.cc60 auto kernel_h = weight_tensor->Height(); in PackWeight() local
72 auto kernel_h = weight_tensor->Height(); in MallocWeightBiasData() local
194 auto kernel_h = weight_tensor->Height(); in Init() local
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc58 …r<int64_t> GetOutputShape(const std::vector<int64_t> &in_shape, int64_t kernel_d, int64_t kernel_h, in GetOutputShape()
92 void GetPadsByPadding(int64_t in_d, int64_t in_h, int64_t in_w, int64_t kernel_d, int64_t kernel_h,… in GetPadsByPadding()
138 auto kernel_h = kernel_size[1]; in InferShape() local
Davg_pool.cc107 auto kernel_h = kernel_size[2]; in InferShape() local
Dmax_pool.cc102 auto kernel_h = kernel_size[2]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32/
Dconv_common_fp32.c351 int kernel_h = conv_param->kernel_h_; in ConvSWFp32() local
433 …nv3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv3x32Kernel()
592 …nv1x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x32Kernel()
687 …nv4x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv4x24Kernel()
862 …nv1x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x24Kernel()
949 …nv6x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv6x16Kernel()
1127 …nv1x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x16Kernel()
1207 …nv12x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv12x8Kernel()
1387 …onv4x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv4x8Kernel()
1484 …onv1x8Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x8Kernel()
[all …]
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()
929 …onvDwCenter(float *dst, const float *src, const float *weight, int height, int width, int kernel_h, in DeconvDwCenter()
1066 int kernel_h = conv_param->kernel_h_; in DepthwiseSWAvxFp32() local
1137 …eSWWxKKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSWWxKKernel()
1198 …SW3x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW3x32Kernel()
1343 …SW1x32Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW1x32Kernel()
1422 …SW4x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW4x24Kernel()
1567 …SW1x24Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW1x24Kernel()
1639 …SW4x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW4x16Kernel()
1753 …SW1x16Kernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in DepthwiseSW1x16Kernel()
[all …]
Dpack_fp32.c35 int kernel_h = conv_param->kernel_h_; in Im2ColPackUnitFp32() local
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()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c32 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackDwUnitFp32() local
82 const int kernel_h = conv_param->kernel_h_; in rolling_im2col_hwc() local
153 const int kernel_h = conv_param->kernel_h_; in rolling_im2row_hwc() local
201 const int kernel_h = conv_param->kernel_h_; in col2im_hwc() local
249 const int kernel_h = conv_param->kernel_h_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc89 int32_t kernel_h = filter_tensor_->Height(); in InitWeightBias() local
263 int kernel_h = conv_param->kernel_h_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c31 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackDwUnitFp16() local
81 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackUnitFp16() local
148 const int kernel_h = conv_param->kernel_h_; in RollingCol2ImPackUnitFp16() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc63 int kernel_h = conv_param->kernel_h_; in TEST_F() local
126 int kernel_h = conv_param->kernel_h_; in TEST_F() local
/third_party/mindspore/mindspore/lite/micro/coder/wrapper/int8/
Dconv_init_int8_wrapper.c22 …Init(int8_t *origin_weight, const int32_t *ori_bias, const int32_t *filter_quant_zps, int kernel_h, in ConvInit()
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc81 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
Dconvolution_fp32_coder.cc59 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
Dconv2d_delegate_fp32_coder.cc77 int kernel_h = conv_param->kernel_h_; in CPUConvolutionFP32CoderSelect() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c55 int kernel_h = param->kernel_h_ != -1 ? param->kernel_h_ : GetHeight(weight); in Deconv2dInferShape() local
Dconv2d_infer.c21 int kernel_h = param->kernel_h_; in ConvInferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dconv_common_base.c46 int kernel_h = conv_param->kernel_h_; in SelectOutputUnit() local
/third_party/mindspore/mindspore/core/ops/fusion/
Dmax_pool_fusion.cc73 auto kernel_h = kernel_size[2]; in InferShape() local
Davg_pool_fusion.cc76 auto kernel_h = kernel_size[2]; 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()
232 …er(float *dst, const float *src, const float *weight, size_t height, size_t width, size_t kernel_h, in DeconvDwFp32Center()

12