Commit 48cd6fa1 authored by Brian Pickrell's avatar Brian Pickrell
Browse files

cosmetic

parent b0b02e63
......@@ -109,9 +109,9 @@ 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.
*
*
* @param inputs list of input shapes
* @return shape
* @return shape
*/
shape normalize_compute_shape(std::vector<shape> inputs) const
{
......@@ -120,7 +120,7 @@ struct reduce_op : op_name<Derived>
if(s.dynamic())
{
auto output_dyn_dims = s.dyn_dims();
auto tuned_axes = tune_axes(output_dyn_dims.size());
auto tuned_axes = tune_axes(output_dyn_dims.size());
for(const auto& axis : tuned_axes)
{
output_dyn_dims[axis] = {1, 1};
......
......@@ -1397,7 +1397,7 @@ void test_reduce_ops()
throws_shape(T{{4}}, input);
}
}
// dynamic shape
template <class T>
void test_dyn_reduce_ops()
......@@ -1405,14 +1405,16 @@ void test_dyn_reduce_ops()
{
migraphx::shape input{migraphx::shape::float_type, {{2, 3, 3}, {2, 4, 4}}};
expect_shape(migraphx::shape{migraphx::shape::float_type,
std::vector<migraphx::shape::dynamic_dimension>({{2, 3, 3}, {1, 1, 0}})},
std::vector<migraphx::shape::dynamic_dimension>(
{{2, 3, 3}, {1, 1, 0}})},
T{{-1}},
input);
}
{
migraphx::shape input{migraphx::shape::float_type, {{2, 3, 3}, {2, 4, 4}}};
expect_shape(migraphx::shape{migraphx::shape::float_type,
std::vector<migraphx::shape::dynamic_dimension>({{1, 1, 0}, {2, 4, 4}})},
std::vector<migraphx::shape::dynamic_dimension>(
{{1, 1, 0}, {2, 4, 4}})},
T{{0}},
input);
}
......
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