Home
last modified time | relevance | path

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

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dconv_sw_avx_fp32.c22size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv3x32AVXKernel()
181size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x32AVXKernel()
276size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x24AVXKernel()
451size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x24AVXKernel()
538size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv6x16AVXKernel()
716size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x16AVXKernel()
796size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv12x8AVXKernel()
976size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv4x8AVXKernel()
1073size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConv1x8AVXKernel()
1165size_t kernel_w, size_t act_flag, size_t ow_block, size_t oc_block, size_t out_step, in SWConvWxKAVXKernel()
Dcommon_func_fp32.h33 size_t kernel_w; member
Dconv_im2col_fp32.c22 int kernel_w = conv_param->kernel_w_; in Im2ColDataPackUnitFp32() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp32/
Dconvolution_slidewindow_fp32.cc49 int kernel_w = filter_tensor->Width(); in Prepare() local
199 int kernel_w = filter_tensor->Width(); in PackWeight() local
212 int kernel_w = filter_tensor->Width(); in MallocWeightBiasData() local
Dadder_fp32.cc71 int kernel_w = filter_tensor->Width(); 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()
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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dconvolution_slidewindow.c89 int kernel_w = GetWidth(filter_tensor); in ConvSWPackWeight() local
109 int kernel_w = GetWidth(filter_tensor); in ConvSWMallocWeightBiasData() local
212 int kernel_w = GetWidth(filter_tensor); in ConvolutionSWPrepare() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/fp16/
Ddeconvolution_fp16.cc61 auto kernel_w = weight_tensor->Width(); in PackWeight() local
73 auto kernel_w = weight_tensor->Width(); in MallocWeightBiasData() local
241 auto kernel_w = weight_tensor->Width(); 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.cu23 … const 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.c32 const int kernel_w = conv_param->kernel_w_; in RollingIm2ColPackDwUnitFp32() local
82 const int kernel_w = conv_param->kernel_w_; in rolling_im2col_hwc() local
153 const int kernel_w = conv_param->kernel_w_; in rolling_im2row_hwc() local
201 const int kernel_w = conv_param->kernel_w_; in col2im_hwc() local
249 const int kernel_w = conv_param->kernel_w_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Dmax_pooling_coreml.cc38 auto kernel_w = static_cast<int>(*(pooling_prim_->kernel_size()->begin() + 1)); in BuildLayer() local
Davg_pooling_coreml.cc39 auto kernel_w = static_cast<int>(*(pooling_prim_->kernel_size()->begin() + 1)); in BuildLayer() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc97 int32_t kernel_w = filter_tensor_->Width(); in InitWeightBias() local
279 int kernel_w = conv_param->kernel_w_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/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-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()
133 … int64_t kernel_h, int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetPadsByPadding()
201 auto kernel_w = kernel_size[4]; in AvgPool3DInferShape() local
Dmax_pool3d.cc143 … int64_t kernel_d, int64_t kernel_h, int64_t kernel_w, int64_t stride_d, in GetOutputShape()
218 … int64_t kernel_h, int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetPadsByPadding()
267 auto kernel_w = kernel_size[kInputIndex4]; in MaxPool3DInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/
Dmaxpool_fusion_mapper.cc79 int64_t kernel_w = kernel_value[1]; in IsKernelSizeValid() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp16/
Dconv2d_delegate_fp16_coder.cc66 int kernel_w = conv_param->kernel_w_; in SelectOutUnit() local
122 int kernel_w = conv_param->kernel_w_; in CPUConvolutionFP16CoderSelect() local
Dconv_depthwise_sw_fp16_coder.cc49 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
Dconvolution_fp16_coder.cc53 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/common/
Dpack_tests.cc66 int kernel_w = conv_param->kernel_w_; in TEST_F() local
129 int kernel_w = conv_param->kernel_w_; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/
Ddeconv2d_infer.c56 int kernel_w = param->kernel_w_ != -1 ? param->kernel_w_ : GetWidth(weight); in Deconv2dInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/fp32/
Ddeconv2d_fp32_coder.cc80 int kernel_w = filter_tensor_->Width(); in InitWeightBias() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dconv_common_base.c47 int kernel_w = conv_param->kernel_w_; in SelectOutputUnit() local

123