Skip to content

N
ngraph
Commit b0903fe8 authored by Amy Zhuang's avatar Amy Zhuang Committed by Robert Kimball
Browse files
Options

Do not throw out of bound exception for onehot in CODEGEN mode. (#2635)

parent b6387054
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 11 deletions
src/ngraph/runtime/cpu/cpu_emitter.cpp
View file @ b0903fe8
...@@ -1949,14 +1949,10 @@ namespace ngraph ...@@ -1949,14 +1949,10 @@ namespace ngraph
writer.block_end(); writer.block_end();
writer << "size_t pos = pos_raw;\n" writer << "size_t pos = pos_raw;\n"
<< "if (pos >= " << bounds << ")\n"; << "if (pos < " << bounds << ")\n";
writer.block_begin(); writer.block_begin();
writer << "throw(std::range_error(\"One-hot: value is out of category "
"range\"));\n";
writer.block_end();
writer << "out_vector(pos, 0) = 1;\n"; writer << "out_vector(pos, 0) = 1;\n";
writer.block_end();
writer.block_end(); writer.block_end();
} }
...@@ -1983,14 +1979,11 @@ namespace ngraph ...@@ -1983,14 +1979,11 @@ namespace ngraph
writer << "size_t pos = pos_raw;\n"; writer << "size_t pos = pos_raw;\n";
writer << "bool found = false;\n"; writer << "bool found = false;\n";
writer << "if (pos >= " << bounds << ")\n"; writer << "if (pos < " << bounds << ")\n";
writer.block_begin(); writer.block_begin();
writer << "throw(std::range_error(\"One-hot: value is out of category "
"range\"));\n";
writer.block_end();
writer << "out_vector" writer << "out_vector"
<< (oh->get_one_hot_axis() == 0 ? "(pos, i)" : "(i, pos)") << " = 1;\n"; << (oh->get_one_hot_axis() == 0 ? "(pos, i)" : "(i, pos)") << " = 1;\n";
writer.block_end();
writer.block_end(); writer.block_end();
......
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