Function find_input_pointwise

src/fuse_pointwise.cpp:217–241

}

static auto find_input_pointwise(const module& m, instruction_ref ins, bool multi_out)
{
    auto it = std::find_if(ins->inputs().begin(), ins->inputs().end(), [&](auto i) {
        return i->name() == "pointwise" and i->outputs().size() == 1 and m.has_instruction(i);
    });
    if(it == ins->inputs().end() and multi_out)
    {
        it = std::find_if(ins->inputs().begin(), ins->inputs().end(), [&](auto i) {
            if(not m.has_instruction(i))
                return false;
            auto base_distance = std::distance(i, ins);
            return i->name() == "pointwise" and
                   std::none_of(i->outputs().begin(), i->outputs().end(), [&](auto output) {
                       if(not m.has_instruction(output))
                           return true;
                       if(output == ins)
                           return false;
                       if(std::distance(i, output) > base_distance)
                           return false;
                       return reaches(output, ins, &m);
                   });
        });
    }
    return it;
}

static std::vector<instruction_ref>
find_output_pointwise(const module& m, instruction_ref ins, bool multi_out)