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

[POT] Added GNA3.5 to the POT's devices list (#14071)

Browse Source 
* Added GNA3.5 to the special devices list

* Remove cpp device filtration
This commit is contained in:
Nikita Malinin 2022-11-19 04:35:33 +01:00 committed by GitHub
parent 42b816ace7
commit ed643b3f74
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/common/offline_transformations/src/pot_transformations.cpp
View File

@ -13,12 +13,10 @@
bool ngraph::pass::POTTransformations::run_on_model(const std::shared_ptr<ngraph::Function>& f) {
ngraph::pass::Manager manager(get_pass_config());
if (m_device == "GNA") {
manager.register_pass<ngraph::pass::BidirectionalSequenceDecomposition>();
manager.register_pass<ngraph::pass::ConvertSequenceToTensorIterator>();
manager.register_pass<ngraph::pass::GRUCellDecomposition>();
manager.register_pass<ngraph::pass::LSTMCellDecomposition>();
}
manager.register_pass<ngraph::pass::BidirectionalSequenceDecomposition>();
manager.register_pass<ngraph::pass::ConvertSequenceToTensorIterator>();
manager.register_pass<ngraph::pass::GRUCellDecomposition>();
manager.register_pass<ngraph::pass::LSTMCellDecomposition>();
manager.run_passes(f);
return false;
}

2
tools/pot/openvino/tools/pot/graph/graph_utils.py
View File

@ -24,7 +24,7 @@ def load_graph(model_config, target_device='ANY'):
""" Loads model from specified path
:return NetworkX model
"""
special_transform_devices = ['GNA']
special_transform_devices = ['GNA', 'GNA3.5']
serialized_bin_path = os.path.join(tempfile.gettempdir(), 'serialized_ir.bin')
serialized_xml_path = os.path.join(tempfile.gettempdir(), 'serialized_ir.xml')
bin_path = model_config.weights