| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | update_conv2d_param_pass.cc | 28 if (prim->GetAttr(ops::kKernelSize) == nullptr) { in SetConvAttr()
29 prim->AddAttr(ops::kKernelSize, MakeValue(kernel_size)); in SetConvAttr()
31 auto origin_kernel_size = GetValue<std::vector<int64_t>>(prim->GetAttr(ops::kKernelSize)); in SetConvAttr()
33 prim->AddAttr(ops::kKernelSize, MakeValue(kernel_size)); in SetConvAttr()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | conv2d_transpose_fusion.cc | 44 …(void)CheckAndConvertUtils::CheckInteger(kKernelSize, SizeToLong(kernel_size.size()), kEqual, kern… in set_kernel_size()
46 (void)CheckAndConvertUtils::CheckInteger(kKernelSize, item, kGreaterEqual, 1, name()); in set_kernel_size()
48 (void)AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
|
| D | max_pool_fusion.cc | 65 auto kernel_size = GetValue<std::vector<int64_t>>(primitive->GetAttr(kKernelSize)); in InferShape()
|
| D | avg_pool_fusion.cc | 67 auto kernel_size = GetValue<std::vector<int64_t>>(primitive->GetAttr(kKernelSize)); in InferShape()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | pool_grad.cc | 58 std::vector<int64_t> k_size = _grad_check_vector(kKernelSize, kernel_size, this->name()); in set_kernel_size()
59 (void)this->AddAttr(kKernelSize, MakeValue(k_size)); in set_kernel_size()
78 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | conv2d_backprop_input.cc | 38 …CheckAndConvertUtils::CheckAttrIntOrTupleInt("kernel_size", primitive->GetAttr(kKernelSize), prim_… in SetPadList()
195 …(void)AddAttr(kKernelSize, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel… in set_kernel_size()
249 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | de_conv2d_grad_filter.cc | 63 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
67 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | group_conv2d_grad_input.cc | 64 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
68 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | conv2d_backprop_filter.cc | 153 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
157 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d_transpose.cc | 57 …(void)CheckAndConvertUtils::CheckInteger(kKernelSize, SizeToLong(kernel_size.size()), kEqual, kern… in set_kernel_size()
59 (void)CheckAndConvertUtils::CheckInteger(kKernelSize, item, kGreaterEqual, 1, name()); in set_kernel_size()
61 (void)AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
129 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | max_pool.cc | 36 (void)this->AddAttr(kKernelSize, in set_kernel_size()
37 … MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel_size, this->name()))); in set_kernel_size()
40 …> MaxPool::get_kernel_size() const { return GetValue<std::vector<int64_t>>(GetAttr(kKernelSize)); } in get_kernel_size()
94 auto kernel_size = GetValue<std::vector<int64_t>>(primitive->GetAttr(kKernelSize)); in InferShape()
|
| D | avg_pool.cc | 36 (void)this->AddAttr(kKernelSize, in set_kernel_size()
37 … MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel_size, this->name()))); in set_kernel_size()
40 …> AvgPool::get_kernel_size() const { return GetValue<std::vector<int64_t>>(GetAttr(kKernelSize)); } in get_kernel_size()
92 auto kernel_size = GetValue<std::vector<int64_t>>(primitive->GetAttr(kKernelSize)); in InferShape()
|
| D | adder.cc | 51 (void)this->AddAttr(kKernelSize, MakeValue(kernel_size)); in set_kernel_size()
55 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | conv2d.cc | 270 …(void)AddAttr(kKernelSize, MakeValue(CheckAndConvertUtils::CheckPositiveVector(kKernelSize, kernel… in set_kernel_size()
319 auto value_ptr = GetAttr(kKernelSize); in get_kernel_size()
|
| D | avg_pool_3d.cc | 38 *kernel_size = GetValue<std::vector<int64_t>>(primitive->GetAttr(kKernelSize)); in GetAttrs()
|
| D | op_utils.h | 106 constexpr auto kKernelSize = "kernel_size"; variable
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | avgpool_fusion_mapper.cc | 39 ValuePtr val_ptr = src_prim->GetAttr(ops::kKernelSize); in Mapper()
|
| D | primitive_mapper.cc | 139 auto status = AttrAdjust(dst_prim, ops::kKernelSize); in AdjustPoolAttr()
|
| /third_party/mindspore/mindspore/lite/tools/converter/import/ |
| D | primitive_adjust.cc | 192 if (name == ops::kKernelSize && origin_value.size() == 1) { in AttrAdjust()
319 status = AttrAdjust(dst_prim, ops::kKernelSize, {0, 1}); in MoveAttrMapConv2D()
362 auto status = AttrAdjust(dst_prim, ops::kKernelSize, {2, 3}); in MoveAttrPool()
398 auto status = AttrAdjust(dst_prim, ops::kKernelSize, {2, 3}); in MoveAttrPoolGrad()
434 status = AttrAdjust(dst_prim, ops::kKernelSize, {0, 1}); in MoveAttrMapAdder()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_pad_fusion.cc | 82 ValuePtr kernel_node = conv_primitive->GetAttr(ops::kKernelSize); in ReplaceParamsAndNodes()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 223 MS_CHECK_TRUE_RET(prim->GetAttr(ops::kKernelSize) != nullptr, RET_ERROR); in CheckConv2DPrimitiveType()
|
| D | depthwise_conv2d_info.cc | 410 MS_CHECK_TRUE_RET(prim->GetAttr(ops::kKernelSize) != nullptr, RET_ERROR); in CheckDepthWiseConv2DPrimitiveType()
|