Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "e7d31abdf5c226eebbdc699d83be90b8cfb8e710"
Commit cdf1d239 authored by charlie's avatar charlie
Browse files

Fix test

parent 4d59204f
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 3 deletions
test/onnx/onnx_test.cpp
View file @ cdf1d239
...@@ -5699,9 +5699,11 @@ TEST_CASE(squeeze_unsqueeze_dyn_test) ...@@ -5699,9 +5699,11 @@ TEST_CASE(squeeze_unsqueeze_dyn_test)
auto l0 = mm->add_parameter( auto l0 = mm->add_parameter(
"0", "0",
migraphx::shape{migraphx::shape::float_type, migraphx::shape{migraphx::shape::float_type,
{{1, 1, 0}, {1, 4, 0}, {1, 1, 0}, {1, 1, 1}, {1, 4, 0}, {1, 1, 0}}}); {{1, 1, 0}, {1, 4, 0}, {1, 1, 0}, {1, 1, 0}, {1, 4, 0}, {1, 1, 0}}});
auto l1 = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", squeeze_axes}}), l0); auto c0 = mm->add_instruction(migraphx::make_op("contiguous"), l0);
auto ret = mm->add_instruction(migraphx::make_op("unsqueeze", {{"axes", unsqueeze_axes}}), l1); auto l1 = mm->add_instruction(migraphx::make_op("squeeze", {{"axes", squeeze_axes}}), c0);
auto c1 = mm->add_instruction(migraphx::make_op("contiguous"), l1);
auto ret = mm->add_instruction(migraphx::make_op("unsqueeze", {{"axes", unsqueeze_axes}}), c1);
mm->add_return({ret}); mm->add_return({ret});
migraphx::onnx_options options; migraphx::onnx_options options;
......
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