m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

rm empty_tensor inputs for merge

Browse Source
pull/432/head
chenyemeng 5 years ago
parent eb058ac969
commit 17d1e9dac5
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

2
ge/graph/passes/merge_pass.cc
Unescape View File

@ -213,7 +213,7 @@ bool MergePass::IsMergeInputNeedOptimized(NodePtr &node) const {
return true; return true;
} }
Status MergePass::OptimizeEmptyTensorInput(const NodePtr &node) const { Status MergePass::OptimizeEmptyTensorInput(const NodePtr &node) {
for (const auto &in_data_anchor : node->GetAllInDataAnchors()) { for (const auto &in_data_anchor : node->GetAllInDataAnchors()) {
const auto &peer_data_anchor = in_data_anchor->GetPeerOutAnchor(); const auto &peer_data_anchor = in_data_anchor->GetPeerOutAnchor();
if (peer_data_anchor == nullptr) { if (peer_data_anchor == nullptr) {

2
ge/graph/passes/merge_pass.h
Unescape View File

@ -29,7 +29,7 @@ class MergePass : public BaseNodePass {
Status ChangeIndexToConstant(NodePtr &node, int &value_index); Status ChangeIndexToConstant(NodePtr &node, int &value_index);
Status CreateConstByValue(NodePtr &node, int value_index, OpDescPtr &op_desc); Status CreateConstByValue(NodePtr &node, int value_index, OpDescPtr &op_desc);
bool IsMergeInputNeedOptimized(NodePtr &node) const; bool IsMergeInputNeedOptimized(NodePtr &node) const;
static Status OptimizeEmptyTensorInput(const NodePtr &node) const; static Status OptimizeEmptyTensorInput(const NodePtr &node);
}; };
} // namespace ge } // namespace ge
#endif // GE_GRAPH_PASSES_MERGE_PASS_H_ #endif // GE_GRAPH_PASSES_MERGE_PASS_H_

Write Preview
Loading…
Cancel
Save