Browse Source

Make NetOutput shape continuous.

pull/1514/head
zhaozhixuan 4 years ago
parent
commit
ce5a7825b6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/reshape_recovery_pass.cc

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

@@ -73,7 +73,7 @@ Status InsertReshapeIfNeed(const NodePtr &node) {
}
if (dst_node->GetType() == NETOUTPUT && is_dynamic) {
// NetOutput shape must be continuous when dynamic shape.
// Otherwise, there may be an arror waiting for the shape refresh to time out during execution.
// Otherwise, there may be an error waiting for the shape refresh to time out during execution.
dst_tensor->SetShape(src_tensor->GetShape());
continue;
}


Loading…
Cancel
Save