Method exec

dnn/src/fallback/split/opr_impl.cpp:33–44

namespace fallback {

void SplitImpl::exec(
        _megdnn_tensor_in src, _megdnn_out const TensorNDArray& dsts,
        _megdnn_workspace workspace) {
    auto dsts_layout = apply_vector<TensorLayout>(m_get_layout, dsts);
    auto dsts_shape = apply_vector<TensorShape>(m_get_shape, dsts_layout);
    check_exec(src.layout, dsts_layout, workspace.size);
    size_t* Bv = reinterpret_cast<size_t*>(workspace.raw_ptr);
    size_t A, B, C;
    get_ABC(dsts_shape, A, Bv, C);
    B = std::accumulate(Bv, Bv + dsts.size(), 0u);
    MEGDNN_DISPATCH_CPU_KERN_OPR(split::exec_generic(src, dsts, A, B, C, Bv));
}

}  // namespace fallback
}  // namespace megdnn