Skip to content

N
ngraph
Commit 4d3e4721 authored by Jaikrishnan Menon's avatar Jaikrishnan Menon Committed by Jayaram Bobba
Browse files
Options

Fix broken tensor write/bug introduced by a cleanup (#1232)

parent 9bb0b653
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/tools/nbench/benchmark.cpp
View file @ 4d3e4721
......@@ -127,7 +127,7 @@ void init_int_tv(shared_ptr<runtime::TensorView> tv, T min, T max)
{
element = dist(s_random_engine);
}
tv->write(vec.data(), 0, vec.size());
tv->write(vec.data(), 0, vec.size() * sizeof(T));
}
template <typename T>
......@@ -140,7 +140,7 @@ void init_real_tv(shared_ptr<runtime::TensorView> tv, T min, T max)
{
element = dist(s_random_engine);
}
tv->write(vec.data(), 0, vec.size());
tv->write(vec.data(), 0, vec.size() * sizeof(T));
}
static void random_init(shared_ptr<runtime::TensorView> tv)
......
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