Skip to content

N
ngraph
Unverified Commit 7c1ab3ef authored by Robert Kimball's avatar Robert Kimball Committed by GitHub
Browse files
Options

initialize all int32_t with 0 or 1 to make OneHot happt (#1720)

parent 5fc7cf65
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/tools/nbench/benchmark.cpp
View file @ 7c1ab3ef
...@@ -119,7 +119,7 @@ static void random_init(shared_ptr<runtime::Tensor> tv) ...@@ -119,7 +119,7 @@ static void random_init(shared_ptr<runtime::Tensor> tv)
} }
else if (et == element::i32) else if (et == element::i32)
{ {
init_int_tv<int32_t>(tv, -1, 1); init_int_tv<int32_t>(tv, 0, 1);
} }
else if (et == element::i64) else if (et == element::i64)
{ {
......
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