Browse Source

rm empty_tensor inputs for merge

tags/v1.1.0
chenyemeng 4 years ago
parent
commit
17d1e9dac5
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      ge/graph/passes/merge_pass.cc
  2. +1
    -1
      ge/graph/passes/merge_pass.h

+ 1
- 1
ge/graph/passes/merge_pass.cc View File

@@ -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) {


+ 1
- 1
ge/graph/passes/merge_pass.h View File

@@ -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_

Loading…
Cancel
Save