Function preserve_output_layout

src/layout_convolution.cpp:68–91

}

void preserve_output_layout(module& m)
{
    auto last = std::prev(m.end());
    if(last->name() == "@return")
    {
        std::vector<instruction_ref> outputs;
        std::transform(last->inputs().begin(),
                       last->inputs().end(),
                       std::back_inserter(outputs),
                       [&](instruction_ref ins) {
                           if(skip_layout(ins->get_shape()))
                               return ins;
                           auto permutation = find_permutation(ins->get_shape());
                           return m.insert_instruction(
                               last, make_op("layout", {{"permutation", permutation}}), ins);
                       });
        m.replace_return(outputs);
    }
    else if(not skip_layout(last->get_shape()))
    {
        auto permutation = find_permutation(last->get_shape());
        m.add_instruction(make_op("layout", {{"permutation", permutation}}), last);
    }
}

void transform_convolutions(module& m, const layout_convolution& lc)
{