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

Use shape{0} for dynamic tensors

Browse Source
This commit is contained in:
Artur Kulikowski 2023-02-02 12:42:06 +01:00
parent e1c2839a5a
commit 08460a486b
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/core/src/model.cpp
View File

@ -479,7 +479,7 @@ inline ov::Tensor create_tmp_tensor(const ngraph::HostTensorPtr& tensor) {
if (tensor->get_element_type().is_dynamic()) {
return std::move(ov::Tensor());
} else {
return std::move(ov::Tensor(tensor->get_element_type(), {0, std::numeric_limits<size_t>::max()}));
return std::move(ov::Tensor(tensor->get_element_type(), {0}));
}
}
}

2
src/core/src/node.cpp
View File

@ -706,7 +706,7 @@ inline ngraph::HostTensorVector create_tmp_tensors(const ov::TensorVector& tenso
ngraph::HostTensorVector result;
result.reserve(tensors.size());
for (const auto& tensor : tensors) {
if (!tensor || tensor.get_shape() == ov::Shape{0, std::numeric_limits<size_t>::max()}) {
if (!tensor || tensor.get_shape() == ov::Shape{0}) {
auto el_type = ov::element::dynamic;
if (tensor)
el_type = tensor.get_element_type();

2
src/plugins/template/backend/int_executable.cpp
View File

@ -65,7 +65,7 @@ inline ov::TensorVector convert_hosttensors_2_tensors(const HostTensorVector& ho
if (hosttensor->get_element_type().is_dynamic()) {
tensor = ov::Tensor();
} else if (hosttensor->get_partial_shape().is_dynamic()) {
tensor = ov::Tensor(hosttensor->get_element_type(), {0, std::numeric_limits<size_t>::max()});
tensor = ov::Tensor(hosttensor->get_element_type(), {0});
} else {
tensor = ov::Tensor(hosttensor->get_element_type(), hosttensor->get_shape());
if (copy_data) {