"git@developer.sourcefind.cn:gaoqiong/migraphx.git" did not exist on "b93f53203c320c788c19affbe25469080fd447bc"
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 ...@@ -34,7 +34,7 @@ struct test_gather_onnx_axis_one_ex : verify_program<test_gather_onnx_axis_one_e
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); 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}}; migraphx::shape s_indices{migraphx::shape::int32_type, {2, 1}};
std::vector<int> indices{0, 2}; std::vector<int> indices{0, 2};
auto a0 = mm->add_parameter("data", s); 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