From bd9d63861db8cd33943c197690b1ee61d1335e5c Mon Sep 17 00:00:00 2001 From: zhengjun10 Date: Mon, 28 Sep 2020 16:53:08 +0800 Subject: [PATCH] transformat optimize --- .../legacy_optimizer/graph/trans_format_insert_pass.cc | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc b/mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc index 7a0534231f..51c758a1b6 100644 --- a/mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc +++ b/mindspore/lite/tools/converter/legacy_optimizer/graph/trans_format_insert_pass.cc @@ -88,6 +88,11 @@ bool TransOpInsertPass::CanFusion(schema::MetaGraphT *graph, const std::unique_p } auto total_node_count = input_node_indexes.size() + output_node_indexes.size(); size_t half_count = total_node_count / 2; + if (GetCNodeTType(*node) == schema::PrimitiveType_Activation) { + if (node->primitive->value.AsActivation()->type == schema::ActivationType_LEAKY_RELU) { + return has_trans_count >= half_count; + } + } if (total_node_count % 2 == 0) { can_fusion = has_trans_count > half_count; } else {