Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit d7be8aa

Browse files
author
Tomasz Dołbniak
authored
Fix the MSVC build error in ONNX OneHot (NervanaSystems#4356)
1 parent fe054d6 commit d7be8aa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/ngraph/frontend/onnx_import/op/onehot.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,9 @@ namespace ngraph
4242
auto off_on_values =
4343
std::make_shared<default_opset::Split>(values, split_axis, 2);
4444
auto off_value =
45-
reshape::interpret_as_scalar(get_output_element(off_on_values, 0ul));
45+
reshape::interpret_as_scalar(get_output_element(off_on_values, size_t{0}));
4646
auto on_value =
47-
reshape::interpret_as_scalar(get_output_element(off_on_values, 1ul));
47+
reshape::interpret_as_scalar(get_output_element(off_on_values, size_t{1}));
4848

4949
auto axis = node.get_attribute_value<std::int64_t>("axis", -1);
5050

0 commit comments

Comments
 (0)