| 3179 | } |
| 3180 | |
| 3181 | static void test_reduce_ops_variable_axes(const std::string& name) |
| 3182 | { |
| 3183 | { |
| 3184 | migraphx::shape input_shape{migraphx::shape::float_type, {2, 3, 4}}; |
| 3185 | migraphx::shape axes_shape{migraphx::shape::int64_type, {1}}; |
| 3186 | migraphx::shape expected_shape{migraphx::shape::float_type, {{1, 2}, {1, 3}, {1, 4}}}; |
| 3187 | expect_shape(expected_shape, migraphx::make_op(name), input_shape, axes_shape); |
| 3188 | } |
| 3189 | |
| 3190 | { |
| 3191 | migraphx::shape input_shape{migraphx::shape::float_type, {{2, 3}, {3, 4}}}; |
| 3192 | migraphx::shape axes_shape{migraphx::shape::int64_type, {1}}; |
| 3193 | migraphx::shape expected_shape{migraphx::shape::float_type, {{1, 3}, {1, 4}}}; |
| 3194 | expect_shape(expected_shape, migraphx::make_op(name), input_shape, axes_shape); |
| 3195 | } |
| 3196 | } |
| 3197 | |
| 3198 | TEST_CASE(reduce_max) { test_reduce_ops("reduce_max"); } |
| 3199 | TEST_CASE(reduce_min) { test_reduce_ops("reduce_min"); } |
no test coverage detected