From 844eac63cebf1fb86b5f9443bd15d141cf21e185 Mon Sep 17 00:00:00 2001 From: zjhellofss Date: Mon, 18 Mar 2024 22:27:05 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9conv=E4=B8=AD=E7=9A=84nodisca?= =?UTF-8?q?rd?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- source/layer/details/base_convolution.hpp | 4 ++-- source/layer/details/deconvolution.hpp | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/source/layer/details/base_convolution.hpp b/source/layer/details/base_convolution.hpp index 600b6aac..5a046c9a 100644 --- a/source/layer/details/base_convolution.hpp +++ b/source/layer/details/base_convolution.hpp @@ -45,8 +45,8 @@ class BaseConvolutionLayer : public ParamLayer { static StatusCode CreateInstance(const std::shared_ptr& op, std::shared_ptr>& conv_layer); - [[nodiscard]] StatusCode Forward(const std::vector>>& inputs, - std::vector>>& outputs) override; + StatusCode Forward(const std::vector>>& inputs, + std::vector>>& outputs) override; private: virtual void ComputeOutput(sftensor input, sftensor output_tensor, uint32_t kernel_h, diff --git a/source/layer/details/deconvolution.hpp b/source/layer/details/deconvolution.hpp index 4e2dc435..704d1d6c 100644 --- a/source/layer/details/deconvolution.hpp +++ b/source/layer/details/deconvolution.hpp @@ -59,9 +59,9 @@ class DeconvolutionLayer : public BaseConvolutionLayer { uint32_t kernel_count_group, uint32_t kernel_h, uint32_t kernel_w, uint32_t output_h, uint32_t output_w) const; - arma::fmat DeconvGEMM(const sftensor& input, uint32_t input_h, uint32_t input_w, - uint32_t channels_per_group, uint32_t group, uint32_t kernel_index, - uint32_t kernel_count_group) const; + [[nodiscard]] arma::fmat DeconvGEMM(const sftensor& input, uint32_t input_h, uint32_t input_w, + uint32_t channels_per_group, uint32_t group, + uint32_t kernel_index, uint32_t kernel_count_group) const; }; } // namespace kuiper_infer #endif // KUIPER_INFER_SOURCE_LAYER_DETAILS_DECONVOLUTION_H