Method get_workspace_in_bytes

dnn/src/cuda/cond_take/opr_impl.cpp:23–26

}

size_t CondTakeImpl::get_workspace_in_bytes(
        const TensorLayout& data, const TensorLayout&) {
    return make_bundle(data.total_nr_elems()).total_size_in_bytes();
}

CondTakeImpl::Output CondTakeImpl::exec(
        _megdnn_tensor_in data, _megdnn_tensor_in mask, _megdnn_workspace workspace,