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

Added transformations for GNA device (#5409)

Browse Source
This commit is contained in:
Nikita Malinin 2021-04-28 10:52:15 +03:00 committed by GitHub
parent 7d5fd3d51c
commit d13de771af
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
inference-engine/src/offline_transformations/src/pot_transformations.cpp
View File

@ -14,11 +14,8 @@ NGRAPH_RTTI_DEFINITION(ngraph::pass::POTTransformations, "POTTransformations", 0
bool ngraph::pass::POTTransformations::run_on_function(std::shared_ptr<ngraph::Function> f) {
ngraph::pass::Manager manager(get_pass_config());
if (m_device == "CPU") {
// TODO: register CPU passes
// manager.register_pass<ngraph::pass::LSTMCellDecomposition>();
} else {
throw ngraph_error("Device name is unsupported");
if (m_device == "GNA") {
manager.register_pass<ngraph::pass::LSTMCellDecomposition>();
}
manager.run_passes(f);
return false;