| 67 | } |
| 68 | |
| 69 | static migraphx::module create_gather(int64_t axis) |
| 70 | { |
| 71 | migraphx::module m; |
| 72 | migraphx::shape sd{migraphx::shape::float_type, {2, 3, 4}}; |
| 73 | migraphx::shape si{migraphx::shape::int64_type, {2, 3}}; |
| 74 | auto di = m.add_parameter("data", sd); |
| 75 | auto ii = m.add_parameter("ind", si); |
| 76 | auto r = m.add_instruction(migraphx::make_op("gather", {{"axis", axis}}), di, ii); |
| 77 | m.add_return({r}); |
| 78 | |
| 79 | return m; |
| 80 | } |
| 81 | |
| 82 | TEST_CASE(gather_test) |
| 83 | { |
no test coverage detected