IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Remove WA from ImportNetwork (#10111)

Browse Source
This commit is contained in:
Ilya Churaev 2022-02-04 07:16:57 +03:00 committed by GitHub
parent 54678f47cf
commit 7b5a4e8c5e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 46 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
src/inference/src/ie_core.cpp
View File

@ -680,52 +680,6 @@ public:
auto parsed = parseDeviceNameIntoConfig(deviceName, config);
auto exec = GetCPPPluginByName(parsed._deviceName).import_model(networkModel, parsed._config);
if (isNewAPI()) {
// create getInputs() based on GetInputsInfo()
using namespace InferenceEngine::details;
if (exec->getInputs().empty() && !exec->GetInputsInfo().empty()) {
const auto& inputsInfo = exec->GetInputsInfo();
std::vector<std::shared_ptr<const ov::Node>> params;
params.reserve(inputsInfo.size());
for (auto&& input : inputsInfo) {
auto param = std::make_shared<ov::op::v0::Parameter>(
convertPrecision(input.second->getPrecision()),
ov::PartialShape(input.second->getTensorDesc().getDims()));
param->set_friendly_name(input.first);
param->get_output_tensor(0).add_names({input.first});
params.emplace_back(std::move(param));
}
exec->setInputs(params);
}
if (exec->getOutputs().empty()) {
const auto& outputsInfo = exec->GetOutputsInfo();
OPENVINO_ASSERT(!outputsInfo.empty(), "outputsInfo is empty after network import");
std::vector<std::shared_ptr<const ov::Node>> results;
results.reserve(outputsInfo.size());
for (auto&& output : outputsInfo) {
auto fake_param = std::make_shared<ov::op::v0::Parameter>(
convertPrecision(output.second->getPrecision()),
ov::PartialShape(output.second->getTensorDesc().getDims()));
fake_param->set_friendly_name(output.first);
auto result = std::make_shared<ov::op::v0::Result>(fake_param);
result->get_output_tensor(0).add_names({output.first});
results.emplace_back(std::move(result));
}
exec->setOutputs(results);
}
// but for true support plugins need:
// 1. ensure order or parameters and results as in ov::Model
// 2. provide tensor names for inputs and outputs
// 3. precisions for getInputs and getOutputs should be taken from GetInputsInfo / GetOutputsInfo
// not from ngraph. Plugins should use SetExeNetworkInfo
}
return {exec._ptr, exec._so};
}