|
@@ -798,11 +798,17 @@ void SaveCustomCaffeProtoPath() { |
|
|
Status CreateInputsForInference(const ge::Graph &graph, vector<ge::GeTensor> &inputs) { |
|
|
Status CreateInputsForInference(const ge::Graph &graph, vector<ge::GeTensor> &inputs) { |
|
|
auto compute_graph = ge::GraphUtils::GetComputeGraph(graph); |
|
|
auto compute_graph = ge::GraphUtils::GetComputeGraph(graph); |
|
|
GE_CHECK_NOTNULL(compute_graph); |
|
|
GE_CHECK_NOTNULL(compute_graph); |
|
|
|
|
|
int64_t index = 0; |
|
|
for (ge::NodePtr &input_node : compute_graph->GetAllNodes()) { |
|
|
for (ge::NodePtr &input_node : compute_graph->GetAllNodes()) { |
|
|
GE_CHECK_NOTNULL(input_node); |
|
|
GE_CHECK_NOTNULL(input_node); |
|
|
ge::OpDescPtr op = input_node->GetOpDesc(); |
|
|
ge::OpDescPtr op = input_node->GetOpDesc(); |
|
|
GE_CHECK_NOTNULL(op); |
|
|
GE_CHECK_NOTNULL(op); |
|
|
if (op->GetType() == ge::DATA) { |
|
|
if (op->GetType() == ge::DATA) { |
|
|
|
|
|
if (!op->HasAttr(ge::ATTR_NAME_INDEX)) { |
|
|
|
|
|
(void)ge::AttrUtils::SetInt(op, ge::ATTR_NAME_INDEX, index); |
|
|
|
|
|
GELOGD("Set attr index:%ld for data op:%s", index, op->GetName().c_str()); |
|
|
|
|
|
} |
|
|
|
|
|
index++; |
|
|
GELOGI("Data op inputDesc size is: %zu", op->GetAllInputsDesc().size()); |
|
|
GELOGI("Data op inputDesc size is: %zu", op->GetAllInputsDesc().size()); |
|
|
ge::GeTensorDesc tensor = op->GetInputDesc(0); |
|
|
ge::GeTensorDesc tensor = op->GetInputDesc(0); |
|
|
string data_op_name = op->GetName(); |
|
|
string data_op_name = op->GetName(); |
|
|