Home
last modified time | relevance | path

Searched refs:kDilation (Results 1 – 13 of 13) sorted by relevance

/third_party/mindspore/mindspore/core/ops/fusion/
Dconv2d_transpose_fusion.cc53 …(void)CheckAndConvertUtils::CheckInteger(kDilation, SizeToLong(dilation.size()), kEqual, dilation_… in set_dilation()
55 (void)CheckAndConvertUtils::CheckInteger(kDilation, item, kGreaterEqual, 1, name()); in set_dilation()
57 (void)AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
/third_party/mindspore/mindspore/core/ops/
Dconv2d_transpose.cc75 …(void)CheckAndConvertUtils::CheckInteger(kDilation, SizeToLong(dilation.size()), kGreaterEqual, di… in set_dilation()
77 (void)AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
139 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dadder.cc83 void Adder::set_dilation(const std::vector<int64_t> &dilation) { (void)this->AddAttr(kDilation, Mak… in set_dilation()
86 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dconv2d.cc278 …(void)AddAttr(kDilation, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kDilation, dilation, … in set_dilation()
329 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dop_utils.h65 constexpr auto kDilation = "dilation"; variable
/third_party/mindspore/mindspore/core/ops/grad/
Dconv2d_backprop_input.cc40 …CheckAndConvertUtils::CheckAttrIntOrTupleInt("dilation", primitive->GetAttr(kDilation), prim_name); in SetPadList()
203 …(void)AddAttr(kDilation, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kDilation, dilation, … in set_dilation()
261 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dde_conv2d_grad_filter.cc104 (void)this->AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
108 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dgroup_conv2d_grad_input.cc105 (void)this->AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
109 auto value_ptr = GetAttr(kDilation); in get_dilation()
Dconv2d_backprop_filter.cc202 (void)this->AddAttr(kDilation, MakeValue(dilation)); in set_dilation()
206 auto value_ptr = GetAttr(kDilation); in get_dilation()
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/
Dconv2d_fusion_mapper.cc39 status = AttrAdjust(dst_prim, ops::kDilation); in Mapper()
Dconv2d_transpose_fusion_mapper.cc43 auto status = AttrAdjust(dst_prim, ops::kDilation); in Mapper()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc314 status = AttrAdjust(dst_prim, ops::kDilation, {2, 3}); in MoveAttrMapConv2D()
429 status = AttrAdjust(dst_prim, ops::kDilation, {2, 3}); in MoveAttrMapAdder()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Dconv2d_info.cc220 MS_CHECK_TRUE_RET(prim->GetAttr(ops::kDilation) != nullptr, RET_ERROR); in CheckConv2DPrimitiveType()