Searched refs:kInChannel (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | conv2d_backprop_filter_fusion.cc | 50 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
60 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| D | conv2d_backprop_input_fusion.cc | 45 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
53 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| D | conv2d_fusion.cc | 41 void Conv2DFusion::set_in_channel(const int64_t in_channel) { (void)this->AddAttr(kInChannel, MakeV… in set_in_channel()
50 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | update_conv2d_param_pass.cc | 36 …if (prim->GetAttr(ops::kInChannel) == nullptr || GetValue<int64_t>(prim->GetAttr(ops::kInChannel))… in SetConvAttr()
37 prim->AddAttr(ops::kInChannel, MakeValue(in_channel)); in SetConvAttr()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | conv2d_transpose.cc | 46 (void)AddAttr(kInChannel, in set_in_channel()
47 … MakeValue(CheckAndConvertUtils::CheckInteger(kInChannel, in_channel, kGreaterThan, 0, name()))); in set_in_channel()
119 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| D | adder.cc | 36 void Adder::set_in_channel(const int64_t in_channel) { (void)this->AddAttr(kInChannel, MakeValue(in… in set_in_channel()
39 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| D | op_utils.h | 99 constexpr auto kInChannel = "in_channel"; variable
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | de_conv2d_grad_filter.cc | 43 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
47 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| D | group_conv2d_grad_input.cc | 44 (void)this->AddAttr(kInChannel, MakeValue(in_channel)); in set_in_channel()
48 auto value_ptr = GetAttr(kInChannel); in get_in_channel()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | tf_conv_parser.cc | 84 if (prim->GetAttr(ops::kInChannel) != nullptr) { in Parse()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 218 MS_CHECK_TRUE_RET(prim->GetAttr(ops::kInChannel) != nullptr, RET_ERROR); in CheckConv2DPrimitiveType()
|
| D | depthwise_conv2d_info.cc | 407 MS_CHECK_TRUE_RET(prim->GetAttr(ops::kInChannel) != nullptr, RET_ERROR); in CheckDepthWiseConv2DPrimitiveType()
|