Formatting

src/include/migraph/operators.hpp

@@ -433,10 +433,10 @@ struct flatten
         {
             MIGRAPH_THROW("axis for flatten must be less than tensor rank");
         }
-        auto x = std::accumulate(
-            lens.begin(), lens.begin() + axis, std::size_t{1}, std::multiplies<>{});
-        auto y = std::accumulate(
-            lens.begin() + axis, lens.end(), std::size_t{1}, std::multiplies<>{});
+        auto x =
+            std::accumulate(lens.begin(), lens.begin() + axis, std::size_t{1}, std::multiplies<>{});
+        auto y =
+            std::accumulate(lens.begin() + axis, lens.end(), std::size_t{1}, std::multiplies<>{});
         return {inputs.at(0).type(), {x, y}};
     }
     argument compute(context&, shape output_shape, std::vector<argument> args) const

test/op_shape_test.cpp

@@ -5,48 +5,51 @@
 #include <sstream>
 #include "test.hpp"
 
-template<class... Ts>
+template <class... Ts>
 void expect_shape(migraph::shape expected, migraph::operation op, Ts... xs)
 {
     migraph::program p;
     std::vector<migraph::shape> shapes{xs...};
     std::vector<migraph::instruction_ref> args;
-    for(auto&& s:shapes)
+    for(auto&& s : shapes)
         args.push_back(p.add_outline(s));
     p.add_instruction(op, args);
-    if(p.get_shape() != expected) {
+    if(p.get_shape() != expected)
+    {
         std::cout << "FAILED: Incorrect shape for " << op.name() << ": ";
         std::cout << expected << " != " << p.get_shape() << std::endl;
-        for(auto&& s:shapes)
+        for(auto&& s : shapes)
             std::cout << "    " << s << std::endl;
     }
 }
 
-template<class... Ts>
+template <class... Ts>
 void throws_shape(migraph::operation op, Ts... xs)
 {
     migraph::program p;
     std::vector<migraph::shape> shapes{xs...};
     std::vector<migraph::instruction_ref> args;
-    for(auto&& s:shapes)
+    for(auto&& s : shapes)
         args.push_back(p.add_outline(s));
     bool thrown = test::throws([&] { p.add_instruction(op, args); });
-    if(not thrown) {
+    if(not thrown)
+    {
         std::cout << "FAILED: No error found for " << op.name() << ": ";
-        for(auto&& s:shapes)
+        for(auto&& s : shapes)
             std::cout << "    " << s << std::endl;
     }
 }
 
-template<class...>
-struct always_false
-: std::false_type
-{}; 
+template <class...>
+struct always_false : std::false_type
+{
+};
 
-template<class... Ts>
+template <class... Ts>
 void throws_shape(migraph::shape, Ts...)
 {
-    static_assert(always_false<Ts...>{}, "An expected shape should not be passed to throws_shape function");
+    static_assert(always_false<Ts...>{},
+                  "An expected shape should not be passed to throws_shape function");
 }
 
 void batch_norm_inference_shape()
@@ -96,11 +99,8 @@ void contiguous_shape()
 void reshape_shape()
 {
     migraph::shape input{migraph::shape::float_type, {24, 1, 1, 1}};
-    for(auto&& new_shape:std::vector<std::vector<int64_t>>{
-        {8, 3, 1, 1},
-        {1, 3, 4, 2},
-        {1, 3, 4, 2}
-    })
+    for(auto&& new_shape :
+        std::vector<std::vector<int64_t>>{{8, 3, 1, 1}, {1, 3, 4, 2}, {1, 3, 4, 2}})
     {
         std::vector<std::size_t> lens(new_shape.size());
         std::copy(new_shape.begin(), new_shape.end(), lens.begin());
@@ -108,10 +108,7 @@ void reshape_shape()
         expect_shape(output, migraph::reshape{new_shape}, input);
     }
 
-    for(auto&& new_shape:std::vector<std::vector<int64_t>>{
-        {8, 3, 2, 2},
-        {1, 3, -1, -1}
-    })
+    for(auto&& new_shape : std::vector<std::vector<int64_t>>{{8, 3, 2, 2}, {1, 3, -1, -1}})
     {
         throws_shape(migraph::reshape{new_shape}, input);
     }
@@ -120,11 +117,16 @@ void reshape_shape()
 void flatten_shape()
 {
     migraph::shape input{migraph::shape::float_type, {2, 4, 6, 8}};
-    expect_shape(migraph::shape{migraph::shape::float_type, {1, 2*4*6*8}}, migraph::flatten{0}, input);
-    expect_shape(migraph::shape{migraph::shape::float_type, {2, 4*6*8}}, migraph::flatten{1}, input);
-    expect_shape(migraph::shape{migraph::shape::float_type, {2*4, 6*8}}, migraph::flatten{2}, input);
-    expect_shape(migraph::shape{migraph::shape::float_type, {2*4*6, 8}}, migraph::flatten{3}, input);
-    expect_shape(migraph::shape{migraph::shape::float_type, {2*4*6*8, 1}}, migraph::flatten{4}, input);
+    expect_shape(
+        migraph::shape{migraph::shape::float_type, {1, 2 * 4 * 6 * 8}}, migraph::flatten{0}, input);
+    expect_shape(
+        migraph::shape{migraph::shape::float_type, {2, 4 * 6 * 8}}, migraph::flatten{1}, input);
+    expect_shape(
+        migraph::shape{migraph::shape::float_type, {2 * 4, 6 * 8}}, migraph::flatten{2}, input);
+    expect_shape(
+        migraph::shape{migraph::shape::float_type, {2 * 4 * 6, 8}}, migraph::flatten{3}, input);
+    expect_shape(
+        migraph::shape{migraph::shape::float_type, {2 * 4 * 6 * 8, 1}}, migraph::flatten{4}, input);
     throws_shape(migraph::flatten{5}, input);
 }