From 17d1e9dac5b581bbcc9a0648a27297a20808231e Mon Sep 17 00:00:00 2001 From: chenyemeng Date: Thu, 26 Nov 2020 10:46:48 +0800 Subject: [PATCH] rm empty_tensor inputs for merge --- ge/graph/passes/merge_pass.cc | 2 +- ge/graph/passes/merge_pass.h | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ge/graph/passes/merge_pass.cc b/ge/graph/passes/merge_pass.cc index c6bae2a2..a65c32fd 100644 --- a/ge/graph/passes/merge_pass.cc +++ b/ge/graph/passes/merge_pass.cc @@ -213,7 +213,7 @@ bool MergePass::IsMergeInputNeedOptimized(NodePtr &node) const { return true; } -Status MergePass::OptimizeEmptyTensorInput(const NodePtr &node) const { +Status MergePass::OptimizeEmptyTensorInput(const NodePtr &node) { for (const auto &in_data_anchor : node->GetAllInDataAnchors()) { const auto &peer_data_anchor = in_data_anchor->GetPeerOutAnchor(); if (peer_data_anchor == nullptr) { diff --git a/ge/graph/passes/merge_pass.h b/ge/graph/passes/merge_pass.h index c297a86e..464f2172 100755 --- a/ge/graph/passes/merge_pass.h +++ b/ge/graph/passes/merge_pass.h @@ -29,7 +29,7 @@ class MergePass : public BaseNodePass { Status ChangeIndexToConstant(NodePtr &node, int &value_index); Status CreateConstByValue(NodePtr &node, int value_index, OpDescPtr &op_desc); bool IsMergeInputNeedOptimized(NodePtr &node) const; - static Status OptimizeEmptyTensorInput(const NodePtr &node) const; + static Status OptimizeEmptyTensorInput(const NodePtr &node); }; } // namespace ge #endif // GE_GRAPH_PASSES_MERGE_PASS_H_