rm empty_tensor inputs for merge

pull/432/head
chenyemeng 5 years ago
parent eb058ac969
commit 17d1e9dac5

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

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