Commit 29060678 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

fixup! Add two test cases using the given example in the onnx Gather documentation

parent 0d13db6e
......@@ -34,7 +34,7 @@ struct test_gather_onnx_axis_one_ex : verify_program<test_gather_onnx_axis_one_e
{
migraphx::program p;
auto* mm = p.get_main_module();
migraphx::shape s{migraphx::shape::float_type, {3, 2}};
migraphx::shape s{migraphx::shape::float_type, {3, 3}};
migraphx::shape s_indices{migraphx::shape::int32_type, {2, 1}};
std::vector<int> indices{0, 2};
auto a0 = mm->add_parameter("data", s);
......
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