| 119 | } |
| 120 | |
| 121 | static migraphx::module create_reduce_mean(const std::vector<int64_t>& axes) |
| 122 | { |
| 123 | migraphx::module m; |
| 124 | migraphx::shape s{migraphx::shape::float_type, {2, 3, 4, 5}}; |
| 125 | auto si = m.add_parameter("data", s); |
| 126 | auto r = m.add_instruction(migraphx::make_op("reduce_mean", {{"axes", axes}}), si); |
| 127 | m.add_return({r}); |
| 128 | |
| 129 | return m; |
| 130 | } |
| 131 | |
| 132 | TEST_CASE(reduce_mean_test) |
| 133 | { |
no test coverage detected