15 lines
405 B
C++
15 lines
405 B
C++
#include <inference_engine.hpp>
|
|
#include <ngraph/ngraph.hpp>
|
|
#include "hetero/hetero_plugin_config.hpp"
|
|
|
|
int main() {
|
|
InferenceEngine::Core core;
|
|
auto network = core.ReadNetwork("sample.xml");
|
|
auto function = network.getFunction();
|
|
//! [part0]
|
|
for (auto && op : function->get_ops())
|
|
op->get_rt_info()["affinity"] = std::make_shared<ngraph::VariantWrapper<std::string>>("CPU");
|
|
//! [part0]
|
|
return 0;
|
|
}
|