Formatting

95ec8e51 · Scott Thornton · 1110ef29 · 95ec8e51
Commit 95ec8e51 authored Aug 10, 2018 by Scott Thornton
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/include/migraph/operators.hpp src/include/migraph/operators.hpp +2 -2

No files found.
--- a/src/include/migraph/operators.hpp
+++ b/src/include/migraph/operators.hpp
@@ -427,11 +427,11 @@ struct flatten
    shape compute_shape(std::vector<shape> inputs) const
    {
        check_shapes{inputs}.has(1);
-        if (axis == 0)
+        if(axis == 0)
        {
            return {inputs.at(0).type(), {1, inputs.at(0).elements()}};
        }
-        if (axis == 1)
+        if(axis == 1)
        {
            return {inputs.at(0).type(), {inputs.at(0).elements(), 1}};
        }