Home
last modified time | relevance | path

Searched defs:stride_w (Results 1 – 25 of 32) sorted by relevance

12

/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()
51 int stride_w = 1; in TEST_F() local
97 int stride_w = 1; in TEST_F() local
144 int stride_w = 1; in TEST_F() local
188 int stride_w = 1; in TEST_F() local
Dpooling_tests.cc25 …ateParameter(PoolMode pool_mode, int window_h, int window_w, int stride_h, int stride_w, int pad_u, in CreateParameter()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Davg_pooling_npu.cc28 auto stride_w = static_cast<int>(*(pooling_prim->strides()->begin() + 1)); in IsSupport() local
48 auto stride_w = static_cast<int>(*(pooling_prim->strides()->begin() + 1)); in SetPoolingParam() local
Dconvolution_npu.cc29 auto stride_w = static_cast<int>(*(conv_prim->stride()->begin() + 1)); in IsSupport() local
41 auto stride_w = static_cast<int>(*(conv_prim->stride()->begin() + 1)); in SetConvParam() local
Dconvolution_depthwise_npu.cc22 auto stride_w = static_cast<int>(*(conv_prim->stride()->begin() + 1)); in SetConvDwParam() local
Dmax_pooling_npu.cc40 auto stride_w = static_cast<int>(*(pooling_prim->strides()->begin() + 1)); in SetPoolingParam() local
Ddeconvolution_npu.cc45 auto stride_w = static_cast<int>(*(conv_prim->stride()->begin() + 1)); in SetDeconvParam() local
/third_party/mindspore/mindspore/core/ops/
Davg_pool_3d.cc59 … int64_t kernel_w, int64_t stride_d, int64_t stride_h, int64_t stride_w, in GetOutputShape()
93 int64_t stride_d, int64_t stride_h, int64_t stride_w, const int64_t &pad_mode, in GetPadsByPadding()
142 auto stride_w = strides[2]; in InferShape() local
Davg_pool.cc110 auto stride_w = strides[3]; in InferShape() local
Dmax_pool.cc105 auto stride_w = strides[3]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c27 const int stride_w = conv_param->stride_w_; in RollingIm2ColPackDwUnitFp32() local
77 const int stride_w = conv_param->stride_w_; in rolling_im2col_hwc() local
148 const int stride_w = conv_param->stride_w_; in rolling_im2row_hwc() local
196 const int stride_w = conv_param->stride_w_; in col2im_hwc() local
244 const int stride_w = conv_param->stride_w_; in rolling_col2im_hwc() local
Dpooling_grad.c22 int stride_w = pooling_param->stride_w_; in AvgPoolingGrad() local
105 int stride_w = pooling_param->stride_w_; in MaxPoolingGrad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c26 const int stride_w = conv_param->stride_w_; in RollingIm2ColPackDwUnitFp16() local
76 const int stride_w = conv_param->stride_w_; in RollingIm2ColPackUnitFp16() local
143 const int stride_w = conv_param->stride_w_; in RollingCol2ImPackUnitFp16() local
Dpooling_grad.c22 int stride_w = pooling_param->stride_w_; in AvgPoolingFp16Grad() local
107 int stride_w = pooling_param->stride_w_; in MaxPoolingFp16Grad() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc38 int64_t stride_w = strides.at(kIndexW); in ApproximateFLOPs() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c58 int stride_w = param->stride_w_; in Deconv2dInferShape() local
Dconv2d_infer.c22 int stride_w = param->stride_w_; in ConvInferShape() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dconv2d_transpose.cc92 int stride_w = param->stride_w_; in SetGlobalLocal() local
112 int stride_w = param->stride_w_; in SetConstArgs() local
Dstack.cc133 cl_int2 stride_w = {stride_w_out, stride_w_in}; in SetConstArgs() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dpooling_gpu_kernel.h207 int stride_w = stride_[3]; in SetPad() local
252 int stride_w = stride_[4]; in SetPad3D() local
Dpooling_grad_gpu_kernel.h242 int stride_w = stride_[3]; in SetPad() local
293 int stride_w = stride_[4]; in SetPad3D() local
/third_party/mindspore/mindspore/core/ops/fusion/
Dmax_pool_fusion.cc76 auto stride_w = strides[3]; in InferShape() local
Davg_pool_fusion.cc79 auto stride_w = strides[3]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dspace_to_batch_int8.c38 int32_t stride_w = (n / in_shape[0]) % block_w; in DoSpaceToBatchNHWCInt8() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc266 int stride_w = conv_param->stride_w_; in CPUConv2DINT8CoderCreator() local

12