/** * \file dnn/src/naive/rnn_cell/opr_impl.cpp * MegEngine is Licensed under the Apache License, Version 2.0 (the "License") * * Copyright (c) 2014-2021 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. */#include "src/naive/rnn_cell/opr_impl.h"#include "src/common/rnn_cell.h"namespacemegdnn{namespacenaive{size_tRNNCellImpl::get_workspace_in_bytes(constTensorLayout&input,constTensorLayout&weight_ih,constTensorLayout&bias_ih,constTensorLayout&hx,constTensorLayout&weight_hh,constTensorLayout&bias_hh,constTensorLayout&dst){returnmegdnn::rnn_cell::get_workspace_in_bytes(input,weight_ih,bias_ih,hx,weight_hh,bias_hh,dst,handle());}voidRNNCellImpl::exec(_megdnn_tensor_ininput,_megdnn_tensor_inweight_ih,_megdnn_tensor_inbias_ih,_megdnn_tensor_inhx,_megdnn_tensor_inweight_hh,_megdnn_tensor_inbias_hh,_megdnn_tensor_outdst,_megdnn_workspaceworkspace){megdnn::rnn_cell::exec(input,weight_ih,bias_ih,hx,weight_hh,bias_hh,dst,workspace,param().nonlineMode,handle());}}// namespace naive}// namespace megdnn