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

Subgraphs Dumper fixes (#13356)

Browse Source 
* Fixed exception in case more than one output

* Fixed a loosing an operations in case of constant inputs
This commit is contained in:
Georgy Krivoruchko 2022-10-10 19:38:15 +04:00 committed by GitHub
parent 589c95ec19
commit 28954a82d9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/tests/functional/plugin/conformance/subgraphs_dumper/src/op_cloner.cpp
View File

@ -91,6 +91,8 @@ std::shared_ptr<ov::Node> clone(const std::shared_ptr<ov::Node> &node, LayerTest
constant->get_shape());
op_inputs.push_back(param);
has_parameters = true;
} else {
const auto clone = std::make_shared<ov::op::v0::Constant>(constant->get_element_type(),
constant->get_shape(),
@ -143,6 +145,8 @@ std::shared_ptr<ov::Node> clone_weightable_node(const std::shared_ptr<ov::Node>
auto param = std::make_shared<ov::op::v0::Parameter>(constant_input->get_element_type(),
constant_input->get_shape());
op_inputs.push_back(param);
has_parameters = true;
} else {
const auto clone = std::make_shared<ov::op::v0::Constant>(constant_input->get_element_type(),
constant_input->get_shape(),

7
src/tests/functional/plugin/conformance/subgraphs_dumper/src/ops_cache.cpp
View File

@ -48,10 +48,11 @@ void OPCache::update_ops_cache(const std::shared_ptr<ov::Node> &op,
} else {
for (int i = 0; i < op->get_input_size(); i++) {
auto shape = op->get_input_shape(i);
unsigned long shapeSize = ov::shape_size(shape) * op->get_element_type().size();
unsigned long shapeSize = ov::shape_size(shape) * op->get_output_element_type(0).size();
auto cachedOpShape = cachedOp->get_input_shape(i);
unsigned long cachedOpShapeSize = ov::shape_size(cachedOpShape) * cachedOp->get_element_type().size();
unsigned long cachedOpShapeSize =
ov::shape_size(cachedOpShape) * cachedOp->get_output_element_type(0).size();
if (shapeSize < cachedOpShapeSize) {
m_ops_cache.erase(cachedOp);
@ -151,7 +152,7 @@ float OPCache::get_size_of_cached_ops() {
op.first->get_input_node_shared_ptr(i));
if (constant != nullptr) {
size += static_cast<float>(ov::shape_size(constant->get_shape()) *
constant->get_element_type().size()) / (1024 * 1024);
constant->get_output_element_type(0).size()) / (1024 * 1024);
}
}
}