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