Method PoolingForward

src/opr/impl/dnn/pooling.cpp:13–23

MGB_DYN_TYPE_OBJ_FINAL_IMPL(PoolingForward);

PoolingForward::PoolingForward(
        VarNode* i0, const Param& param, const ExecutionPolicy& policy,
        const OperatorNodeConfig& config)
        : Super(OperatorNodeBaseCtorParam{i0->owner_graph(), config, "pooling", {i0}}) {
    init_megdnn_opr(*this, param);
    add_input({i0});
    output(0)->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE);
    m_policy = policy;

    intl::MegDNNOprInitPostCtor<PoolingForward>::apply(*this);
}

SymbolVar PoolingForward::make(
        SymbolVar i0, const Param& param, const ExecutionPolicy& policy,