Function remove_layout

src/layout_convolution.cpp:116–128

}

void remove_layout(module& m)
{
    for(auto ins : iterator_for(m))
    {
        if(ins->name() != "layout")
            continue;
        auto perm  = ins->get_operator().to_value()["permutation"].to_vector<std::int64_t>();
        auto iperm = find_permutation(ins->inputs().front()->get_shape());
        if(perm != iperm)
            continue;
        m.replace_instruction(ins, ins->inputs().front());
    }
}
} // namespace

void layout_convolution::apply(module_pass_manager& mpm) const