Format

23490657 · Paul · c164d8e6 · 23490657
Commit 23490657 authored Oct 28, 2022 by Paul
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 2 deletions

src/layout_nhwc.cpp src/layout_nhwc.cpp +3 -2

No files found.
--- a/src/layout_nhwc.cpp
+++ b/src/layout_nhwc.cpp
@@ -32,8 +32,9 @@ std::vector<instruction_ref> find_lasts(const module& m, Predicate pred)
 std::unordered_set<instruction_ref> preserve_output_layout(module& m)
 {
    std::unordered_set<instruction_ref> result;
-    std::vector<instruction_ref> outputs =
+    std::vector<instruction_ref> outputs = find_lasts(m, [](auto ins) {
-        find_lasts(m, [](auto ins) { return ins->name() == "convolution" and ins->get_shape().lens().size() == 4; });
+        return ins->name() == "convolution" and ins->get_shape().lens().size() == 4;
+    });
    for(auto output : outputs)
    {
        auto permutation = find_permutation(output->get_shape());