Method exec

dnn/src/x86/pooling/algo.cpp:105–124

}

void PoolingImpl::AlgoMeanW2S2AVX::exec(const ExecArgs& args) const {
    auto N = args.layout_src.shape[0];
    auto C = args.layout_src.shape[1];
    auto IH = args.layout_src.shape[2];
    auto IW = args.layout_src.shape[3];
    auto OH = args.layout_dst.shape[2];
    auto OW = args.layout_dst.shape[3];
    auto PH = args.opr->param().pad_h;
    auto PW = args.opr->param().pad_w;

    auto handle = [=]() { return args.handle; };
    MEGDNN_DISPATCH_CPU_KERN_OPR(
            auto sptr = reinterpret_cast<dt_float32*>(args.src_tensor.raw_ptr());
            auto dptr = reinterpret_cast<dt_float32*>(args.dst_tensor.raw_ptr());
            rep(n, N) rep(c, C) {
                mean_pooling_w2x2_s2x2_avx(
                        sptr + n * C * IH * IW + c * IH * IW, IH, IW,
                        dptr + n * C * OH * OW + c * OH * OW, OH, OW, PH, PW, true);
            });
}

bool PoolingImpl::AlgoMeanW2S2SSE3::is_available(const SizeArgs& args) const {
    auto SH = args.opr->param().stride_h;