|
|
|
@ -182,7 +182,7 @@ STATUS CaffeModelParser::ConvertGraphInputs() {
|
|
|
|
|
auto type_ptr = TypeIdToType(TypeId::kNumberTypeFloat32);
|
|
|
|
|
auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape);
|
|
|
|
|
parameter->set_abstract(abstract_tensor);
|
|
|
|
|
parameter->set_name("graph-input-" + std::to_string(i));
|
|
|
|
|
parameter->set_name("graph_input-" + std::to_string(i));
|
|
|
|
|
nodes_.insert(std::pair(layer.top(0), parameter));
|
|
|
|
|
return RET_OK;
|
|
|
|
|
}
|
|
|
|
@ -205,7 +205,7 @@ STATUS CaffeModelParser::ConvertGraphInputs() {
|
|
|
|
|
auto type_ptr = TypeIdToType(TypeId::kNumberTypeFloat32);
|
|
|
|
|
auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape);
|
|
|
|
|
parameter->set_abstract(abstract_tensor);
|
|
|
|
|
parameter->set_name("graph-input-" + caffe_model_.input(i));
|
|
|
|
|
parameter->set_name("graph_input-" + caffe_model_.input(i));
|
|
|
|
|
nodes_.insert(std::pair(caffe_model_.input(i), parameter));
|
|
|
|
|
}
|
|
|
|
|
} else {
|
|
|
|
@ -219,7 +219,7 @@ STATUS CaffeModelParser::ConvertGraphInputs() {
|
|
|
|
|
auto type_ptr = TypeIdToType(TypeId::kNumberTypeFloat32);
|
|
|
|
|
auto abstract_tensor = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector);
|
|
|
|
|
parameter->set_abstract(abstract_tensor);
|
|
|
|
|
parameter->set_name("graph-input-" + caffe_model_.input(i));
|
|
|
|
|
parameter->set_name("graph_input-" + caffe_model_.input(i));
|
|
|
|
|
nodes_.insert(std::pair(caffe_model_.input(i), parameter));
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|