Searched refs:kPadMode (Results 1 – 20 of 20) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | max_pool.cc | 31 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
34 PadMode MaxPool::get_pad_mode() const { return PadMode(GetValue<int64_t>(GetAttr(kPadMode))); } in get_pad_mode()
95 auto pad_mode_value = (primitive->GetAttr(kPadMode)); in InferShape()
|
| D | avg_pool.cc | 31 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
34 PadMode AvgPool::get_pad_mode() const { return PadMode(GetValue<int64_t>(GetAttr(kPadMode))); } in get_pad_mode()
93 auto pad_mode = PadMode(GetValue<int64_t>(primitive->GetAttr(kPadMode))); in InferShape()
|
| D | adder.cc | 61 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
65 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | conv2d_transpose.cc | 90 (void)AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
144 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | conv2d.cc | 291 (void)AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
334 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | avg_pool_3d.cc | 53 CheckAndConvertUtils::GetPadModEnumValue(primitive->GetAttr(kPadMode), pad_mode, true); in GetAttrs()
|
| D | op_utils.h | 157 constexpr auto kPadMode = "pad_mode"; variable
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | pool_grad.cc | 69 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
90 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | pooling_grad.cc | 64 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
68 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | de_conv2d_grad_filter.cc | 74 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
78 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | group_conv2d_grad_input.cc | 75 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
79 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | conv2d_backprop_input.cc | 44 CheckAndConvertUtils::GetPadModEnumValue(primitive->GetAttr(kPadMode), &pad_mode, true); in SetPadList()
216 (void)AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
267 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| D | conv2d_backprop_filter.cc | 164 (void)this->AddAttr(kPadMode, MakeValue(swi)); in set_pad_mode()
168 auto value_ptr = GetAttr(kPadMode); in get_pad_mode()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | max_pool_fusion.cc | 66 auto pad_mode = PadMode(GetValue<int64_t>(primitive->GetAttr(kPadMode))); in InferShape()
|
| D | avg_pool_fusion.cc | 68 auto pad_mode = PadMode(GetValue<int64_t>(primitive->GetAttr(kPadMode))); in InferShape()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_conv_fusion.cc | 224 …int64_t up_pad_mode = up_conv_primitive->GetAttr(ops::kPadMode) == nullptr ? 0 : up_conv_primitive… in IsPrimitiveProper()
226 … down_conv_primitive->GetAttr(ops::kPadMode) == nullptr ? 0 : down_conv_primitive->get_pad_mode(); in IsPrimitiveProper()
|
| D | conv_pad_fusion.cc | 69 …int64_t conv_pad_mode = conv_primitive->GetAttr(ops::kPadMode) == nullptr ? 0 : conv_primitive->ge… in ReplaceParamsAndNodes()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | primitive_mapper.cc | 117 auto pad_mode_val = dst_prim->GetAttr(ops::kPadMode); in AdjustOnnxPoolAttr()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | multi_conv_info.cc | 70 MS_CHECK_TRUE_RET(conv_prim->GetAttr(ops::kPadMode) != nullptr, false); in CheckSplitValid()
|
| /third_party/mindspore/mindspore/core/utils/ |
| D | check_convert_utils.cc | 94 {ops::kPadMode, PadModeConverter},
99 {ops::kPadMode, PadModeUpperConverter},
|