Skip to content

Commit

Permalink
[onert] Revisit ConvolutionLayer.h (#11443)
Browse files Browse the repository at this point in the history
It makes methods private as much as possible.
Also, it reorders the methods to make it easy to read.

ONE-DCO-1.0-Signed-off-by: Sanggyu Lee <[email protected]>
  • Loading branch information
glistening authored Sep 1, 2023
1 parent 913150c commit 1aa0919
Showing 1 changed file with 6 additions and 9 deletions.
15 changes: 6 additions & 9 deletions runtime/onert/backend/cpu/ops/ConvolutionLayer.h
Original file line number Diff line number Diff line change
Expand Up @@ -48,24 +48,21 @@ class ConvolutionLayer : public ::onert::exec::IFunction
~ConvolutionLayer();

public:
void convFloat32();

void convQ8uPerTensor();
void convQ8uPerChannel();

void convQ8i();

void configure(const IPortableTensor *input, const IPortableTensor *kernel,
const IPortableTensor *bias, ir::PaddingType _paddingType,
const uint32_t paddingLeft, const uint32_t paddingRight, const uint32_t paddingTop,
const uint32_t paddingBottom, const uint32_t strideWidth,
const uint32_t strideHeight, const uint32_t dilationWidthFactor,
const uint32_t dilationHeightFactor, const ir::Activation activation,
IPortableTensor *output);

void prepare() override;
void run() override;

void prepare() override;
private:
void convFloat32();
void convQ8uPerTensor();
void convQ8uPerChannel();
void convQ8i();

private:
const IPortableTensor *_input;
Expand Down

0 comments on commit 1aa0919

Please sign in to comment.