Home
last modified time | relevance | path

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

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dconv_sw_avx_fp32.c21 …x32AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv3x32AVXKernel()
180 …x32AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x32AVXKernel()
275 …x24AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv4x24AVXKernel()
450 …x24AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x24AVXKernel()
537 …x16AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv6x16AVXKernel()
715 …x16AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x16AVXKernel()
795 …2x8AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv12x8AVXKernel()
975 …4x8AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv4x8AVXKernel()
1072 …1x8AVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConv1x8AVXKernel()
1164 …WxKAVXKernel(float *dst, const float *src, const float *weight, const float *bias, size_t kernel_h, in SWConvWxKAVXKernel()
Dconv_im2col_fp32.c23 int kernel_h = conv_param->kernel_h_; in Im2ColDataPackUnitFp32() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_slidewindow_fp32.cc48 int kernel_h = filter_tensor->Height(); in Prepare() local
198 int kernel_h = filter_tensor->Height(); in PackWeight() local
211 int kernel_h = filter_tensor->Height(); in MallocWeightBiasData() local
Dadder_fp32.cc70 int kernel_h = filter_tensor->Height(); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconvolution_slidewindow.c88 int kernel_h = GetHeight(filter_tensor); in ConvSWPackWeight() local
108 int kernel_h = GetHeight(filter_tensor); in ConvSWMallocWeightBiasData() local
211 int kernel_h = GetHeight(filter_tensor); in ConvolutionSWPrepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Ddeconvolution_fp16.cc60 auto kernel_h = weight_tensor->Height(); in PackWeight() local
72 auto kernel_h = weight_tensor->Height(); in MallocWeightBiasData() local
240 auto kernel_h = weight_tensor->Height(); in Prepare() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Ddeformable_offsets_impl.cu68 __global__ void GenPositionGridKernel(const uint kernel_h, const uint kernel_w, const uint stride_h, in GenPositionGridKernel()
92 … const uint input_w, const uint kernel_h, const uint kernel_w, const uint num, in DeformableOffsetsKernel()
129 … uint input_h, uint input_w, uint dfm_group, uint kernel_h, uint kernel_w, uint output_h, in DeformableOffsets()
151 cudaError_t GenPositionGrid(const uint kernel_h, const uint kernel_w, const uint stride_h, const ui… in GenPositionGrid()
Davg_pool3d_helper_impl.cu23const int64_t kernel_h, const int64_t kernel_w, const int64_t edge_kernel_d, in RealKernelSize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dpack_ext.c31 const int kernel_h = conv_param->kernel_h_; in RollingIm2ColPackDwUnitFp32() local
81 const int kernel_h = conv_param->kernel_h_; in rolling_im2col_hwc() local
152 const int kernel_h = conv_param->kernel_h_; in rolling_im2row_hwc() local
200 const int kernel_h = conv_param->kernel_h_; in col2im_hwc() local
248 const int kernel_h = conv_param->kernel_h_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dmax_pooling_coreml.cc37 auto kernel_h = static_cast<int>(*(pooling_prim_->kernel_size()->begin())); in BuildLayer() local
Davg_pooling_coreml.cc38 auto kernel_h = static_cast<int>(*(pooling_prim_->kernel_size()->begin())); in BuildLayer() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc96 int32_t kernel_h = filter_tensor_->Height(); in InitWeightBias() local
278 int kernel_h = conv_param->kernel_h_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/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-src/source/mindspore/core/ops/
Davg_pool_3d.cc81 … int64_t kernel_d, int64_t kernel_h, int64_t kernel_w, int64_t stride_d, in GetOutputShape()
133int64_t kernel_h, int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetPadsByPadding()
200 auto kernel_h = kernel_size[3]; in AvgPool3DInferShape() local
Dmax_pool3d.cc143 … int64_t kernel_d, int64_t kernel_h, int64_t kernel_w, int64_t stride_d, in GetOutputShape()
218int64_t kernel_h, int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetPadsByPadding()
266 auto kernel_h = kernel_size[kInputIndex3]; in MaxPool3DInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dmaxpool_fusion_mapper.cc78 int64_t kernel_h = kernel_value[0]; in IsKernelSizeValid() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconv2d_delegate_fp16_coder.cc65 int kernel_h = conv_param->kernel_h_; in SelectOutUnit() local
121 int kernel_h = conv_param->kernel_h_; in CPUConvolutionFP16CoderSelect() local
Dconv_depthwise_sw_fp16_coder.cc48 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
Dconvolution_fp16_coder.cc52 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc65 int kernel_h = conv_param->kernel_h_; in TEST_F() local
128 int kernel_h = conv_param->kernel_h_; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Ddeconv2d_infer.c57 int kernel_h = param->kernel_h_ != -1 ? param->kernel_h_ : GetHeight(weight); in Deconv2dInferShape() local
Dconv2d_infer.c22 int kernel_h = param->kernel_h_; in ConvInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc79 int kernel_h = filter_tensor_->Height(); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dconv_common_base.c46 int kernel_h = conv_param->kernel_h_; in SelectOutputUnit() local

123