/** * \file dnn/src/cuda/concat/opr_impl.h * MegEngine is Licensed under the Apache License, Version 2.0 (the "License") * * Copyright (c) 2014-2020 Megvii Inc. All rights reserved. * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. */#pragma once#include "megdnn/oprs.h"namespacemegdnn{namespacecuda{classConcatForwardImpl:publicConcatForward{public:usingConcatForward::ConcatForward;voidexec(_megdnn_inconstTensorNDArray&srcs,_megdnn_tensor_outdst,_megdnn_workspaceworkspace)override;size_tget_workspace_in_bytes(constTensorLayoutArray&,constTensorLayout&)override;private:template<typenameT>voidexec_internal(_megdnn_inconstTensorNDArray&srcs,_megdnn_tensor_outdst,_megdnn_workspaceworkspace);};}// namespace cuda}// namespace megdnn// vim: syntax=cpp.doxygen