Home
last modified time | relevance | path

Searched defs:stride_h (Results 1 – 25 of 34) 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()
50 int stride_h = 1; in TEST_F() local
96 int stride_h = 1; in TEST_F() local
143 int stride_h = 1; in TEST_F() local
187 int stride_h = 1; in TEST_F() local
Dpooling_tests.cc25 OpParameter *CreateParameter(PoolMode pool_mode, int window_h, int window_w, int stride_h, int stri… in CreateParameter()
/third_party/mindspore/mindspore/lite/src/delegate/npu/op/
Davg_pooling_npu.cc27 auto stride_h = static_cast<int>(*(pooling_prim->strides()->begin())); in IsSupport() local
47 auto stride_h = static_cast<int>(*(pooling_prim->strides()->begin())); in SetPoolingParam() local
Dconvolution_npu.cc28 auto stride_h = static_cast<int>(*(conv_prim->stride()->begin())); in IsSupport() local
40 auto stride_h = static_cast<int>(*(conv_prim->stride()->begin())); in SetConvParam() local
Dconvolution_depthwise_npu.cc21 auto stride_h = static_cast<int>(*(conv_prim->stride()->begin())); in SetConvDwParam() local
Dmax_pooling_npu.cc39 auto stride_h = static_cast<int>(*(pooling_prim->strides()->begin())); in SetPoolingParam() local
Ddeconvolution_npu.cc44 auto stride_h = static_cast<int>(*(conv_prim->stride()->begin())); 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()
141 auto stride_h = strides[1]; in InferShape() local
Davg_pool.cc109 auto stride_h = strides[2]; in InferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dpack_ext.c26 const int stride_h = conv_param->stride_h_; in RollingIm2ColPackDwUnitFp32() local
76 const int stride_h = conv_param->stride_h_; in rolling_im2col_hwc() local
147 const int stride_h = conv_param->stride_h_; in rolling_im2row_hwc() local
195 const int stride_h = conv_param->stride_h_; in col2im_hwc() local
243 const int stride_h = conv_param->stride_h_; in rolling_col2im_hwc() local
Dpooling_grad.c23 int stride_h = pooling_param->stride_h_; in AvgPoolingGrad() local
106 int stride_h = pooling_param->stride_h_; in MaxPoolingGrad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/
Dbatch_to_space_base.c26 int stride_h = block_w * out_n; in BatchToSpaceNoCropForNHWC() local
67 int stride_h = block_w * out_n; in BatchToSpaceForNHWC() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dpack_fp16_ext.c25 const int stride_h = conv_param->stride_h_; in RollingIm2ColPackDwUnitFp16() local
75 const int stride_h = conv_param->stride_h_; in RollingIm2ColPackUnitFp16() local
142 const int stride_h = conv_param->stride_h_; in RollingCol2ImPackUnitFp16() local
Dpooling_grad.c23 int stride_h = pooling_param->stride_h_; in AvgPoolingFp16Grad() local
108 int stride_h = pooling_param->stride_h_; in MaxPoolingFp16Grad() local
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dsplit_strategy.cc37 int64_t stride_h = strides.at(kIndexH); in ApproximateFLOPs() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dbatch_to_space_int8.c26 size_t stride_h = block_w * out_n; in BatchToSpaceNoCropForNHWCInt8() local
72 size_t stride_h = block_w * out_n; in BatchToSpaceForNHWCInt8() local
Dspace_to_batch_int8.c39 int32_t stride_h = (n / in_shape[0]) / block_w; in DoSpaceToBatchNHWCInt8() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Ddeconv2d_infer.c59 int stride_h = param->stride_h_; in Deconv2dInferShape() local
Dconv2d_infer.c23 int stride_h = param->stride_h_; in ConvInferShape() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dpooling_gpu_kernel.h206 int stride_h = stride_[2]; in SetPad() local
251 int stride_h = stride_[3]; in SetPad3D() local
Dpooling_grad_gpu_kernel.h241 int stride_h = stride_[2]; in SetPad() local
292 int stride_h = stride_[3]; in SetPad3D() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dconv2d_transpose.cc91 int stride_h = param->stride_h_; in SetGlobalLocal() local
111 int stride_h = param->stride_h_; in SetConstArgs() local
/third_party/mindspore/mindspore/core/ops/fusion/
Dmax_pool_fusion.cc75 auto stride_h = strides[2]; in InferShape() local
Davg_pool_fusion.cc78 auto stride_h = strides[2]; in InferShape() local
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc265 int stride_h = conv_param->stride_h_; in CPUConv2DINT8CoderCreator() local

12