Home
last modified time | relevance | path

Searched defs:dilation_h (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Ddepthwise_conv2d_tests.cc27 … int pad_r, int dilation_h, int dilation_w, ActType act_type, int input_channel) { in CreateParameter()
56 int dilation_h = 1; in TEST_F() local
102 int dilation_h = 1; in TEST_F() local
149 int dilation_h = 1; in TEST_F() local
193 int dilation_h = 1; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32_grad/
Dpack_ext.c28 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackDwUnitFp32() local
78 const int dilation_h = conv_param->dilation_h_; in rolling_im2col_hwc() local
149 const int dilation_h = conv_param->dilation_h_; in rolling_im2row_hwc() local
197 const int dilation_h = conv_param->dilation_h_; in col2im_hwc() local
245 const int dilation_h = conv_param->dilation_h_; in rolling_col2im_hwc() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/coreml/op/
Ddeconvolution_coreml.cc45 auto dilation_h = static_cast<int>(*(deconv_prim_->dilation()->begin())); in SetConvParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp16_grad/
Dpack_fp16_ext.c28 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackDwUnitFp16() local
78 const int dilation_h = conv_param->dilation_h_; in RollingIm2ColPackUnitFp16() local
145 const int dilation_h = conv_param->dilation_h_; in RollingCol2ImPackUnitFp16() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/delegate/npu/op/
Dconvolution_depthwise_int8_npu.cc39 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvDwParam() local
Dconvolution_depthwise_npu.cc39 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvDwParam() local
Ddeconvolution_npu.cc53 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetDeconvParam() local
Dconvolution_int8_npu.cc52 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvParam() local
Dconvolution_npu.cc48 auto dilation_h = static_cast<int>(*(conv_prim->dilation()->begin())); in SetConvParam() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dconv_im2col_fp32.c26 int dilation_h = conv_param->dilation_h_; in Im2ColDataPackUnitFp32() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Ddeformable_offsets_grad_gpu_kernel.h58 uint dilation_h; member
Dconv3d_grad_input_gpu_kernel.h321 auto dilation_h = dilation[kIndex3]; in GetSameModePadList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Ddeformable_offsets_grad_cpu_kernel.h47 size_t dilation_h = 0; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Ddeformable_offsets_grad.h43 size_t dilation_h = 0; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dim2col_proto.cc149 int64_t dilation_h = dilation[0]; in IMPLEMT_COMMON_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/add_attr_to_node/
Dconv3d_backprop_input_padlist_fusion.cc57 auto dilation_h = dilation[kIndex3]; in SetConv3DBackpropPadList() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dmaxpool_with_argmax_v2_impl.cu74 … const int pads_h, const int pads_w, const int dilation_h, const int dilation_w, in CalMaxPoolWithArgmaxV2()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Ddeformable_offsets_fusion.cc47 int64_t dilation_h = dilations[axis_h]; in CreateHelperNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/
Dconv2d_int8_coder.cc282 int dilation_h = conv_param->dilation_h_; in CPUConv2DINT8CoderCreator() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dim2col.cc127 int64_t dilation_h = dilations.front(); in Im2ColInferShape() local
Dconv3d.cc335 int64_t dilation_h = dilation[kIndex1]; in CaculateShape() local
421 int64_t dilation_h = dilation[kIndex1]; in CheckPadList() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Ddilation2d_backprop_filter.cc129 auto dilation_h = dilation[h_axis]; in Dilation2DBackpropFilterInferShape() local
Ddilation2d_backprop_input.cc129 auto dilation_h = dilation[h_axis]; in Dilation2DBackpropInputInferShape() local
Dconv2d_backprop_filter.cc47 const auto dilation_h = dilation[kIndex2]; in CalcPadListForSameMode() local
Dconv2d_backprop_input.cc50 const auto dilation_h = dilation[kIndex2]; in CalPadListForSameMode() local

12