Method perform

src/opr/impl/nn_int.cpp:167–198

}

void ElemwiseMultiType::perform(
        Mode mode, DType out_dt, DeviceTensorND& dest,
        const SmallVector<DeviceTensorND>& inputs,
        intl::UniqPtrWithCN<megdnn::ElemwiseMultiType>& opr) {
    megdnn::TensorNDArray dnn_inputs(inputs.size());
    TensorShapeArray inp_shapes(inputs.size());
    CompNode out_cn;
    for (size_t i = 0; i < inputs.size(); ++i) {
        auto&& t = inputs[i];
        if (!i) {
            out_cn = t.comp_node();
        } else {
            mgb_assert(t.comp_node() == out_cn);
        }
        if (t.shape().is_empty()) {
            mgb_assert(dest.empty());
            return;
        }
        inp_shapes[i] = t.shape();
    }
    if (!opr) {
        opr = intl::create_megdnn_opr<megdnn::ElemwiseMultiType>(out_cn);
    } else {
        mgb_assert(out_cn == opr.comp_node());
    }
    out_cn.activate();
    for (size_t i = 0; i < inputs.size(); ++i)
        dnn_inputs[i] = inputs[i].as_megdnn();
    dest.comp_node(out_cn).dtype(out_dt).resize(get_output_var_shape(mode, inp_shapes));
    opr->param() = {mode};
    call_megdnn_opr_exec(out_cn, dnn_inputs, dest.as_megdnn(), opr.get(), nullptr);
}

// vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}