Function get_reduce_size

src/split_reduce.cpp:169–174

}

static std::size_t get_reduce_size(const_module_ref rm)
{
    auto ins = std::find_if(rm->begin(), rm->end(), &is_reduce);
    assert(ins != rm->end());
    return ins->inputs().front()->get_shape().elements() / ins->get_shape().elements();
}

void split_reduce::apply(module_pass_manager& mpm) const
{