| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | conv_activation_fusion.cc | 49 …(act_prim->GetAttr(ops::kActivationType) != nullptr && act_prim->get_activation_type() != mindspor… in Process()
67 if (prim->GetAttr(ops::kActivationType) == nullptr || in Process()
68 … static_cast<mindspore::ActivationType>(GetValue<int64_t>(prim->GetAttr(ops::kActivationType))) == in Process()
71 prim->AddAttr(ops::kActivationType, MakeValue<int64_t>(type)); in Process()
|
| D | conv_tuple_activation_fusion.cc | 55 if (act_prim->GetAttr(ops::kActivationType) == nullptr || in Process()
78 …if (primc->GetAttr(ops::kActivationType) == nullptr || primc->get_activation_type() == mindspore::… in Process()
|
| D | tf_gelu_fusion.cc | 42 return act_prim->GetAttr(ops::kActivationType) != nullptr && in CheckTanh()
43 GetValue<int64_t>(act_prim->GetAttr(ops::kActivationType)) == mindspore::TANH; in CheckTanh()
|
| D | sigmoid_mul_fusion.cc | 55 if (activation_prim == nullptr || (activation_prim->GetAttr(ops::kActivationType) != nullptr && in Process()
|
| D | conv_biasadd_fusion.cc | 107 auto add_act_ptr = prim_add->GetAttr(ops::kActivationType); in CheckCanFusion()
125 auto conv_act_ptr = prim_add->GetAttr(ops::kActivationType); in CheckCanFusion()
|
| /third_party/mindspore/mindspore/core/ops/fusion/ |
| D | div_fusion.cc | 27 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
31 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | mul_fusion.cc | 28 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
31 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | sub_fusion.cc | 27 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
31 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | scale_fusion.cc | 31 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
35 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | l2_normalize_fusion.cc | 32 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
36 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | adder_fusion.cc | 40 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
44 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | conv2d_backprop_filter_fusion.cc | 46 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
54 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | activation.cc | 34 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
53 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | add_fusion.cc | 30 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
33 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | conv2d_backprop_input_fusion.cc | 50 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
59 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | conv2d_fusion.cc | 47 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
60 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | conv2d_transpose_fusion.cc | 71 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
81 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | full_connection.cc | 47 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
50 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | avg_pool_fusion.cc | 38 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
48 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | max_pool_fusion.cc | 38 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
48 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | activation_grad.cc | 36 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
40 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | de_conv2d_grad_filter.cc | 134 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
138 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| D | group_conv2d_grad_input.cc | 145 (void)this->AddAttr(kActivationType, MakeValue(swi)); in set_activation_type()
149 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | affine.cc | 40 (void)this->AddAttr(kActivationType, MakeValue(static_cast<int64_t>(activation_type))); in set_activation_type()
64 auto value_ptr = GetAttr(kActivationType); in get_activation_type()
|
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/ |
| D | pass_registry_test.cc | 66 …if (primc->GetAttr(ops::kActivationType) != nullptr && primc->get_activation_type() != mindspore::… in CanFusion()
83 if (add_primc->GetAttr(ops::kActivationType) != nullptr && in CanFusion()
|