Method apply

src/simplify_reshapes.cpp:502–555

    }

    void apply(module& m, const match::matcher_result& mr) const
    {
        auto ins      = mr.result;
        auto slice    = mr.instructions["slice"];
        auto slice_op = slice->get_operator().to_value();
        auto axes     = slice_op.at("axes").to_vector<std::size_t>();

        if(ins->get_shape().scalar())
            return;

        std::vector<operation> ops;
        auto x = ins;
        while(contains(shape_transform_ops(), x->get_operator().name()))
        {
            ops.push_back(x->get_operator());
            x = x->inputs().front();
        }
        if(x != slice)
            return;
        x = x->inputs().front();
        std::reverse(ops.begin(), ops.end());
        auto desc = shape_transform_descriptor::create(slice->get_shape().lens(), ops);

        std::vector<std::size_t> new_axes;
        std::transform(axes.begin(),
                       axes.end(),
                       join_back_inserter(new_axes),
                       [&](auto axis) -> std::vector<std::size_t> {
                           auto result = desc.get_dst_axes_from_src(axis);
                           if(result.size() != 1)
                               return {};
                           return result;
                       });

        // Optimizes shape transforms if the slice cant be optimized
        if(axes.size() != new_axes.size())
        {
            auto opt_ops = desc.generate();
            auto y       = insert_ops(m, ins, opt_ops, slice);
            m.replace_instruction(ins, y);
            return;
        }
        slice_op["axes"] = new_axes;

        auto new_desc = desc.rebase(slice->inputs().front()->get_shape().lens());
        if(new_desc.empty())
            return;
        new_desc.simplify();

        auto opt_ops = new_desc.generate();
        auto y       = insert_ops(m, ins, opt_ops, x);
        y            = m.insert_instruction(ins, make_op("slice", slice_op), y);
        m.replace_instruction(ins, y);
    }
};

struct find_nop_reshapes