Browse Source

ReshapeRemovePass_not_remove_reshape_in_dynamic_subgraph

pull/1911/head
lichun 4 years ago
parent
commit
d8be4eae75
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc

+ 2
- 2
ge/ge_local_engine/ops_kernel_store/ge_local_ops_kernel_builder.cc View File

@@ -22,6 +22,7 @@
#include "graph/utils/node_utils.h"
#include "graph/utils/tensor_utils.h"
#include "graph/utils/type_utils.h"
#include "graph/compute_graph.h"
#include "ge_local_engine/ops_kernel_store/op/op_factory.h"
#include "ge_local_engine/common/constant/constant.h"
#include "register/ops_kernel_builder_registry.h"
@@ -164,8 +165,7 @@ Status GeLocalOpsKernelBuilder::CalcConstantStrMemSize(const OpDescPtr &op_desc,
Status GeLocalOpsKernelBuilder::GenerateTask(const Node &node, RunContext &context, std::vector<domi::TaskDef> &tasks) {
bool is_in_unknown_subgraph = false;
bool forced_unknown = false;
ComputeGraphPtr owner_graph = node.GetOwnerComputeGraph();
for (const auto &node : owner_graph->GetDirectNode()) {
for (const auto &node : node.GetOwnerComputeGraph()->GetDirectNode()) {
GE_CHK_GRAPH_STATUS_RET(ge::NodeUtils::GetNodeUnknownShapeStatus(*node, is_in_unknown_subgraph),
"[Get][ShapeStatus] of node[%s] failed!", node->GetName().c_str());
if (is_in_unknown_subgraph) {


Loading…
Cancel
Save