Browse Source

Feature: reset shape of dynamic single op

pull/563/head
l00444296 4 years ago
parent
commit
eec8af0d70
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/passes/dynamic_single_op_reset_shape_pass.cc

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

@@ -27,7 +27,7 @@ Status DynamicSingleOpResetShapePass::Run(ComputeGraphPtr graph) {
GE_CHECK_NOTNULL(graph);
for (const auto &node : graph->GetDirectNode()) {
GE_CHECK_NOTNULL(node->GetOpDesc());
if (node->GetType() != DATA || node->GetType() == NETOUTPUT) {
if (node->GetType() == DATA || node->GetType() == NETOUTPUT) {
continue;
}



Loading…
Cancel
Save