Skip to content

N
ngraph
Commit ddef598c authored by Michal Karzynski's avatar Michal Karzynski
Browse files
Options

[ONNX] Move CPU tests to INTERPRETER

parent b5446d87
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
test/onnx_import.cpp
View file @ ddef598c
...@@ -1660,7 +1660,7 @@ TEST(onnx, model_argmax_int32) ...@@ -1660,7 +1660,7 @@ TEST(onnx, model_argmax_int32)
std::vector<std::int64_t>{1, 1, 1, 1, 1, 1}}; std::vector<std::int64_t>{1, 1, 1, 1, 1, 1}};
std::vector<std::vector<std::int64_t>> outputs{ std::vector<std::vector<std::int64_t>> outputs{
execute<std::int32_t, std::int64_t>(function, inputs, "CPU")}; execute<std::int32_t, std::int64_t>(function, inputs, "INTERPRETER")};
EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front())); EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front()));
} }
...@@ -1675,6 +1675,6 @@ TEST(onnx, model_argmin_int32) ...@@ -1675,6 +1675,6 @@ TEST(onnx, model_argmin_int32)
std::vector<std::vector<std::int64_t>> expected_output{std::vector<std::int64_t>{0, 0, 0, 0}}; std::vector<std::vector<std::int64_t>> expected_output{std::vector<std::int64_t>{0, 0, 0, 0}};
std::vector<std::vector<std::int64_t>> outputs{ std::vector<std::vector<std::int64_t>> outputs{
execute<std::int32_t, std::int64_t>(function, inputs, "CPU")}; execute<std::int32_t, std::int64_t>(function, inputs, "INTERPRETER")};
EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front())); EXPECT_TRUE(test::all_close(expected_output.front(), outputs.front()));
} }
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