Home
last modified time | relevance | path

Searched refs:pad_mode_ (Results 1 – 25 of 51) sorted by relevance

123

/third_party/mindspore/mindspore/lite/test/ut/nnacl/infer/
Dpooling_infer_test.cc42 parameter->pad_mode_ = Pad_pad; in TEST_F()
82 parameter->pad_mode_ = Pad_pad; in TEST_F()
88 parameter->pad_mode_ = Pad_same; in TEST_F()
123 parameter->pad_mode_ = Pad_pad; in TEST_F()
129 parameter->pad_mode_ = Pad_valid; in TEST_F()
164 parameter->pad_mode_ = Pad_pad; in TEST_F()
170 parameter->pad_mode_ = Pad_valid; in TEST_F()
205 parameter->pad_mode_ = Pad_pad; in TEST_F()
211 parameter->pad_mode_ = Pad_pad; in TEST_F()
246 parameter->pad_mode_ = Pad_pad; in TEST_F()
[all …]
Ddepthwise_conv2d_infer_test.cc143 parameter->pad_mode_ = Pad_pad; in TEST_F()
190 parameter->pad_mode_ = Pad_pad; in TEST_F()
237 parameter->pad_mode_ = Pad_pad; in TEST_F()
284 parameter->pad_mode_ = Pad_pad; in TEST_F()
331 parameter->pad_mode_ = Pad_pad; in TEST_F()
378 parameter->pad_mode_ = Pad_pad; in TEST_F()
425 parameter->pad_mode_ = Pad_same; in TEST_F()
472 parameter->pad_mode_ = Pad_pad; in TEST_F()
519 parameter->pad_mode_ = Pad_same; in TEST_F()
Dconv2d_infer_test.cc140 parameter->pad_mode_ = Pad_pad; in TEST_F()
186 parameter->pad_mode_ = Pad_pad; in TEST_F()
232 parameter->pad_mode_ = Pad_pad; in TEST_F()
278 parameter->pad_mode_ = Pad_pad; in TEST_F()
324 parameter->pad_mode_ = Pad_pad; in TEST_F()
370 parameter->pad_mode_ = Pad_pad; in TEST_F()
416 parameter->pad_mode_ = Pad_same; in TEST_F()
462 parameter->pad_mode_ = Pad_pad; in TEST_F()
508 parameter->pad_mode_ = Pad_same; in TEST_F()
Ddeconv2d_infer_test.cc52 parameter->pad_mode_ = Pad_pad; in TEST_F()
100 parameter->pad_mode_ = Pad_pad; in TEST_F()
148 parameter->pad_mode_ = Pad_pad; in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dim2col_gpu_kernel.h68 if ((pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) && use_pad_) { in Launch()
119 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
121 if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) { in Init()
124 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
184 …if ((pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) && use_pad_ && !is_… in InitSizeLists()
290 std::string pad_mode_; variable
Dpooling_grad_gpu_kernel.h223 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in SetPad()
252 if (kSamePadModeUpperCase == pad_mode_ || kSamePadModeLowerCase == pad_mode_) { in SetPad()
260 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in SetPad()
272 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in SetPad3D()
305 if (kSamePadModeUpperCase == pad_mode_ || kSamePadModeLowerCase == pad_mode_) { in SetPad3D()
316 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in SetPad3D()
352 std::string pad_mode_; variable
Dpooling_gpu_kernel.h185 pad_mode_ = GetValue<std::string>(prim->GetAttr("pad_mode")); in SetPad()
208 if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) { in SetPad()
228 pad_mode_ = GetValue<std::string>(prim->GetAttr("pad_mode")); in SetPad3D()
253 if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) { in SetPad3D()
281 std::string pad_mode_; variable
Dmaxpool_with_argmax_gpu_kernel.h130 pad_mode_ = GetValue<std::string>(prim->GetAttr("pad_mode")); in Init()
133 if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) { in Init()
164 std::string pad_mode_; variable
Dconv3d_transpose_gpu_kernel.h147 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
152 …if (pad_mode_ == kSamePadModeUpperCase || pad_mode_ == kSamePadModeLowerCase) { // pad_mode_ = sa… in Init()
169 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
527 std::string pad_mode_; variable
Dconv3d_gpu_kernel.h122 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
158 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
364 std::string pad_mode_; variable
Dconv2d_gpu_kernel.h120 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
156 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
367 std::string pad_mode_; variable
Dconv3d_grad_input_gpu_kernel.h123 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
159 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
373 std::string pad_mode_; variable
Dconv3d_grad_filter_gpu_kernel.h146 pad_mode_ = GetAttr<std::string>(kernel_node, "pad_mode"); in Init()
182 if (pad_mode_ == kValidPadModeUpperCase || pad_mode_ == kValidPadModeLowerCase) { in Init()
398 std::string pad_mode_; variable
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/
Dmaxpool_info.cc46 pad_mode_ = GetIntAttr(PAD_MODE); in GetAttrs()
47 if (pad_mode_ != POOL_PAD_MODE_VALID && pad_mode_ != POOL_PAD_MODE_SAME) { in GetAttrs()
48 MS_LOG(ERROR) << name_ << ": The pad_mode value is invalid: " << pad_mode_; in GetAttrs()
72 …MS_LOG(INFO) << name_ << ": The kernel size is " << kernel_size_ << ", pad mode is " << pad_mode_ … in GetAttrs()
Dconv2d_info.cc93 pad_mode_ = GetIntAttr(PAD_MODE); in GetAttrsBase()
94 if (pad_mode_ < 0 || pad_mode_ > 2) { in GetAttrsBase()
95 … MS_LOG(ERROR) << name_ << ": The pad_mode must be in the range of [0, 2], but got " << pad_mode_; in GetAttrsBase()
130 … << ", mode is " << mode_ << ", pad mode is " << pad_mode_ << ", pad list is " << pad_list_ in GetAttrsBase()
237 if (pad_mode_ == 0) { // 'pad' mode in CheckHWStrategy()
242 if (pad_mode_ == 1) { // 'same' mode in CheckHWStrategy()
246 if (pad_mode_ == 2) { // 'valid' mode in CheckHWStrategy()
937 if (pad_mode_ != 1) { // only support same mode in CheckHWStrategy()
938 …MS_LOG(ERROR) << name_ << ": Do not support the pad mode " << pad_mode_ << " when split H or W dim… in CheckHWStrategy()
/third_party/mindspore/mindspore/lite/src/ops/populate/v0/
Ddeconv2d_populate_v0.cc67 conv_param->pad_mode_ = Pad_same; in PopulateDeconvParameter()
70 conv_param->pad_mode_ = Pad_valid; in PopulateDeconvParameter()
73 conv_param->pad_mode_ = Pad_pad; in PopulateDeconvParameter()
Ddedepthwise_conv2d_populate_v0.cc67 conv_param->pad_mode_ = Pad_same; in PopulateDeconvDwParameter()
70 conv_param->pad_mode_ = Pad_valid; in PopulateDeconvDwParameter()
73 conv_param->pad_mode_ = Pad_pad; in PopulateDeconvDwParameter()
Ddepthwise_conv2d_populate_v0.cc57 conv_param->pad_mode_ = Pad_same; in PopulateConvDwParameter()
60 conv_param->pad_mode_ = Pad_valid; in PopulateConvDwParameter()
63 conv_param->pad_mode_ = Pad_pad; in PopulateConvDwParameter()
Dconv2d_populate_v0.cc59 conv_param->pad_mode_ = Pad_same; in PopulateConvParameter()
62 conv_param->pad_mode_ = Pad_valid; in PopulateConvParameter()
65 conv_param->pad_mode_ = Pad_pad; in PopulateConvParameter()
Dpooling_populate_v0.cc89 pooling_param->pad_mode_ = Pad_same; in PopulatePoolingParameter()
92 pooling_param->pad_mode_ = Pad_valid; in PopulatePoolingParameter()
95 pooling_param->pad_mode_ = Pad_pad; in PopulatePoolingParameter()
/third_party/mindspore/mindspore/lite/src/ops/populate/
Dconv2d_populate.cc72 param->pad_mode_ = Pad_same; in PopulateConvParameter()
75 param->pad_mode_ = Pad_valid; in PopulateConvParameter()
78 param->pad_mode_ = Pad_pad; in PopulateConvParameter()
Ddeconv2d_populate.cc85 param->pad_mode_ = Pad_same; in PopulateDeconvParameter()
88 param->pad_mode_ = Pad_valid; in PopulateDeconvParameter()
91 param->pad_mode_ = Pad_pad; in PopulateDeconvParameter()
Dpooling_populate.cc51 param->pad_mode_ = Pad_same; in UpdatePadMode()
54 param->pad_mode_ = Pad_valid; in UpdatePadMode()
57 param->pad_mode_ = Pad_pad; in UpdatePadMode()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_populate_parameter.cc202 pooling_param->pad_mode_ = Pad_same; in PopulateMaxPoolGradParameter()
205 pooling_param->pad_mode_ = Pad_valid; in PopulateMaxPoolGradParameter()
208 pooling_param->pad_mode_ = Pad_pad; in PopulateMaxPoolGradParameter()
240 pooling_param->pad_mode_ = Pad_same; in PopulateAvgPoolGradParameter()
243 pooling_param->pad_mode_ = Pad_valid; in PopulateAvgPoolGradParameter()
246 pooling_param->pad_mode_ = Pad_pad; in PopulateAvgPoolGradParameter()
253 pooling_param->pad_mode_ = Pad_same; in PopulateAvgPoolGradParameter()
256 pooling_param->pad_mode_ = Pad_valid; in PopulateAvgPoolGradParameter()
259 pooling_param->pad_mode_ = Pad_pad; in PopulateAvgPoolGradParameter()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/op/
Dpool_tensorrt.cc129 pad_mode_ = pool_primitive->pad_mode(); in ParseParams()
162 pad_mode_ = pool_primitive->pad_mode(); in ParseParams()
177 if (pad_mode_ == schema::PadMode::PadMode_SAME) { in AddParams()

123