Skip to content

N
ngraph
Commit 49a0ae55 authored by pawelpiotrowicz's avatar pawelpiotrowicz Committed by Scott Cyphers
Browse files
Options

no_fork : Remove allocation for empty vector (#3544) 

* no_fork : Remove allication for empty vector

* style
parent d6f933d5
Hide whitespace changes
Inline Side-by-side
Showing with 20 additions and 18 deletions
src/ngraph/op/constant.hpp
View file @ 49a0ae55
...@@ -95,37 +95,39 @@ namespace ngraph ...@@ -95,37 +95,39 @@ namespace ngraph
", expected ", ", expected ",
shape_size(m_shape), shape_size(m_shape),
"."); ".");
if (values.size())
if (type.is_integral())
{ {
if (type.is_signed()) if (type.is_integral())
{ {
std::vector<int64_t> dvalues = parse_string<int64_t>(values); if (type.is_signed())
if (values.size() == 1 && shape_size(m_shape) != 1)
{ {
dvalues = std::vector<int64_t>(shape_size(m_shape), dvalues[0]); std::vector<int64_t> dvalues = parse_string<int64_t>(values);
if (values.size() == 1 && shape_size(m_shape) != 1)
{
dvalues = std::vector<int64_t>(shape_size(m_shape), dvalues[0]);
}
write_values(dvalues);
}
else
{
std::vector<uint64_t> dvalues = parse_string<uint64_t>(values);
if (values.size() == 1 && shape_size(m_shape) != 1)
{
dvalues = std::vector<uint64_t>(shape_size(m_shape), dvalues[0]);
}
write_values(dvalues);
} }
write_values(dvalues);
} }
else else
{ {
std::vector<uint64_t> dvalues = parse_string<uint64_t>(values); std::vector<double> dvalues = parse_string<double>(values);
if (values.size() == 1 && shape_size(m_shape) != 1) if (values.size() == 1 && shape_size(m_shape) != 1)
{ {
dvalues = std::vector<uint64_t>(shape_size(m_shape), dvalues[0]); dvalues = std::vector<double>(shape_size(m_shape), dvalues[0]);
} }
write_values(dvalues); write_values(dvalues);
} }
} }
else
{
std::vector<double> dvalues = parse_string<double>(values);
if (values.size() == 1 && shape_size(m_shape) != 1)
{
dvalues = std::vector<double>(shape_size(m_shape), dvalues[0]);
}
write_values(dvalues);
}
constructor_validate_and_infer_types(); constructor_validate_and_infer_types();
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment