Method exec

dnn/src/arm_common/pooling/opr_impl.cpp:161–183

}

void PoolingImpl::exec(
        _megdnn_tensor_in src, _megdnn_tensor_out dst, _megdnn_workspace workspace) {
    check_exec(src.layout, dst.layout, workspace.size);
    auto param = make_pooling_kern_param(this, src, dst, workspace);

    bool fwd_reduce = is_global_pooling_reduce(param);
    if (fwd_reduce) {
        auto global_pooling_fwd = [=]() {
            auto reduce_tensor = get_gloabl_pooling_reduce_tensor(src, dst);
            auto&& opr = get_global_pooling_reduce_opr(nullptr, param);
            opr->exec(reduce_tensor.first, reduce_tensor.second, workspace);
        };
        MEGDNN_DISPATCH_CPU_KERN_OPR(global_pooling_fwd());
        return;
    }

    auto algo = get_algorithm(this, src.layout, dst.layout);
    if (!is_fallback_algo(algo)) {
        algo->exec(param);
    } else {
        fallback::PoolingImpl::exec(src, dst, workspace);
    }
}

MEGDNN_DEF_GET_ALGO_FROM_DESC(PoolingImpl);