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

modified: ge/graph/passes/replace_with_empty_const_pass.cc

Browse Source 
modified:   ge/graph/passes/replace_with_empty_const_pass.h
pull/460/head
zhaoxinxin 4 years ago
parent 60476ce26e
commit 6e8581d0a6
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File

5
ge/graph/passes/replace_with_empty_const_pass.cc
Unescape View File

@ -74,9 +74,10 @@ Status ReplaceWithEmptyConstPass::Run(NodePtr &node) {
GELOGD("ReplaceWithEmptyConstPass end."); GELOGD("ReplaceWithEmptyConstPass end.");
return SUCCESS; return SUCCESS;
} }
Status GetOutputsOfCurrNode(const NodePtr &node_to_repalce, vector<GeTensorPtr> &outputs) { Status GetOutputsOfCurrNode(const NodePtr &node_to_replace, vector<GeTensorPtr> &outputs) {
for (const auto &out_anchor : node_to_replace->GetAllOutDataAnchors()) { for (const auto &out_anchor : node_to_replace->GetAllOutDataAnchors()) {
auto out_desc = op_desc->GetOutputDesc(out_anchor->GetIdx()); GE_CHECK_NOTNULL(node_to_replace->GetOpDesc());
auto out_desc = node_to_replace->GetOpDesc()->GetOutputDesc(out_anchor->GetIdx());
GeTensorPtr empty_tensor = MakeShared<ge::GeTensor>(out_desc); GeTensorPtr empty_tensor = MakeShared<ge::GeTensor>(out_desc);
GE_CHECK_NOTNULL(empty_tensor); GE_CHECK_NOTNULL(empty_tensor);
outputs.emplace_back(empty_tensor); outputs.emplace_back(empty_tensor);

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

@ -25,7 +25,7 @@ class ReplaceWithEmptyConstPass : public FoldingPass {
Status Run(NodePtr &node) override; Status Run(NodePtr &node) override;
private: private:
Status GetOutputsOfCurrNode(const NodePtr &node_to_repalce, vector<GeTensorPtr> &outputs); Status GetOutputsOfCurrNode(const NodePtr &node_to_replace, vector<GeTensorPtr> &outputs);
bool IsEmptyTenor(const GeShape &shape) const; bool IsEmptyTenor(const GeShape &shape) const;
}; };
} // namespace ge } // namespace ge

Write Preview
Loading…
Cancel
Save