Commit 36f76f90 authored by Brian Pickrell's avatar Brian Pickrell
Browse files

formatting

parent 57d000b2
......@@ -108,7 +108,7 @@ struct reduce_op : op_name<Derived>
/**
* @brief returns a shape in which the axis or axes named
* for reduction by this op are set to size 1.
* for reduction by this op are set, to size 1.
*
* @param inputs list of input shapes
* @return shape
......
......@@ -141,7 +141,7 @@ struct shape
* Same as the number of indices required to get a data value.
*/
std::size_t ndim() const;
/*!
* Return the number of elements in the tensor.
*/
......
......@@ -4241,13 +4241,14 @@ TEST_CASE(reducel1_test)
EXPECT(p == prog);
}
TEST_CASE(reducel1_dyn_test)
{
migraphx::program p;
auto* mm = p.get_main_module();
// a shape with 4 dynamic dimensions
auto l0 = mm->add_parameter("x", migraphx::shape{migraphx::shape::float_type, {{3,3,0}, {3, 5, 0}, {4, 6, 5}, {5, 7, 6}}});
auto l0 = mm->add_parameter(
"x",
migraphx::shape{migraphx::shape::float_type, {{3, 3, 0}, {3, 5, 0}, {4, 6, 5}, {5, 7, 6}}});
auto abs_l0 = mm->add_instruction(migraphx::make_op("abs"), l0);
auto sum_l0 = mm->add_instruction(migraphx::make_op("reduce_sum", {{"axes", {-2}}}), abs_l0);
mm->add_instruction(migraphx::make_op("squeeze", {{"axes", {-2}}}), sum_l0);
......
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