mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!37859 fix cle pattern 

Merge pull request !37859 from yeyunpeng2020/master_cle
This commit is contained in:
i-robot 2022-07-12 11:49:54 +00:00 committed by Gitee
parent 46136386ae c59b2f3611
commit 8eda6b33a9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mindspore/lite/tools/converter/quantizer/cle_pattern.cc
View File

@ -35,7 +35,7 @@ std::unordered_map<std::string, VectorRef> CLEPattern::DefinePatterns() const {
bool IsLinearActivation(const api::SharedPtr<ops::Conv2DFusion> &conv2d) { bool IsLinearActivation(const api::SharedPtr<ops::Conv2DFusion> &conv2d) {
std::set<ActivationType> liner_activations = {RELU, NO_ACTIVATION}; std::set<ActivationType> liner_activations = {RELU, NO_ACTIVATION};
auto value_ptr = conv2d->GetAttr(ops::kActivation); auto value_ptr = conv2d->GetAttr(ops::kActivationType);
if (value_ptr == nullptr || liner_activations.find(conv2d->get_activation_type()) != liner_activations.end()) { if (value_ptr == nullptr || liner_activations.find(conv2d->get_activation_type()) != liner_activations.end()) {
return true; return true;
} }
@ -58,13 +58,13 @@ bool IsConvNode(const BaseRef &n, ConvNode node) {
if (conv == nullptr) { if (conv == nullptr) {
return false; return false;
} }
if (!IsLinearActivation(conv)) {
return false;
}
if (node == COMMON_CONV) { if (node == COMMON_CONV) {
return conv->get_group() == 1; return conv->get_group() == 1;
} else if (node == DEPTHWISE_CONV) { } else if (node == DEPTHWISE_CONV) {
bool ret = IsLinearActivation(conv) && conv->GetAttr(ops::kIsDepthWise) != nullptr && return conv->GetAttr(ops::kIsDepthWise) != nullptr && GetValue<bool>(conv->GetAttr(ops::kIsDepthWise));
GetValue<bool>(conv->GetAttr(ops::kIsDepthWise));
return ret;
} else { } else {
MS_LOG(ERROR) << "Not supported conv node type."; MS_LOG(ERROR) << "Not supported conv node type.";
return false; return false;