Skip to content

Commit 8dd56aa

Browse files
committed
tidy
1 parent dc1647b commit 8dd56aa

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

include/layers/ReduceLayer.hpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
#pragma once
2+
#include <cstdint>
23
#include <vector>
34

45
#include "layers/Layer.hpp"
@@ -8,7 +9,7 @@ namespace it_lab_ai {
89

910
class ReduceLayer : public Layer {
1011
public:
11-
enum class Operation { kSum, kMean, kMult, kMax, kMin } uint8_t;
12+
enum class Operation : uint8_t { kSum, kMean, kMult, kMax, kMin };
1213

1314
ReduceLayer(Operation op, int64_t keepdims = 0);
1415
explicit ReduceLayer(int64_t keepdims = 0)

0 commit comments

Comments
 (0)