Skip to content

Commit

Permalink
修改conv中的nodiscard
Browse files Browse the repository at this point in the history
  • Loading branch information
zjhellofss committed Mar 18, 2024
1 parent e834b5b commit 844eac6
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
4 changes: 2 additions & 2 deletions source/layer/details/base_convolution.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -45,8 +45,8 @@ class BaseConvolutionLayer : public ParamLayer {
static StatusCode CreateInstance(const std::shared_ptr<RuntimeOperator>& op,
std::shared_ptr<Layer<float>>& conv_layer);

[[nodiscard]] StatusCode Forward(const std::vector<std::shared_ptr<Tensor<float>>>& inputs,
std::vector<std::shared_ptr<Tensor<float>>>& outputs) override;
StatusCode Forward(const std::vector<std::shared_ptr<Tensor<float>>>& inputs,
std::vector<std::shared_ptr<Tensor<float>>>& outputs) override;

private:
virtual void ComputeOutput(sftensor input, sftensor output_tensor, uint32_t kernel_h,
Expand Down
6 changes: 3 additions & 3 deletions source/layer/details/deconvolution.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -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

0 comments on commit 844eac6

Please sign in to comment.