Browse Source

mark original input

tags/v1.5.1
wjm 3 years ago
parent
commit
c7f3b08445
2 changed files with 2 additions and 1 deletions
  1. +1
    -0
      ge/generator/ge_generator.cc
  2. +1
    -1
      metadef

+ 1
- 0
ge/generator/ge_generator.cc View File

@@ -205,6 +205,7 @@ static Status AddInputs(const ComputeGraphPtr &graph, const NodePtr &node, const
}

(void)AttrUtils::SetBool(data_op, "_is_single_op", true);
(void)AttrUtils::SetBool(data_op, ATTR_NAME_IS_ORIGINAL_INPUT, true);

GE_CHK_BOOL_EXEC(data_op->AddInputDesc(tensor) == GRAPH_SUCCESS,
REPORT_CALL_ERROR("E19999", "AddInputDesc failed for node:%s", data_op->GetName().c_str());


+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 310610e5392e01659d214ad380e9ed2c39f9f5a3
Subproject commit 2ad00e17886fd06c0d00f8a8cf370783a3d31818

Loading…
Cancel
Save