Commit e014144c authored by Luxios22's avatar Luxios22 Committed by binmakeswell
Browse files

fix format (#565)

parent 1762ba14
...@@ -13,16 +13,14 @@ ...@@ -13,16 +13,14 @@
#include "cublas_wrappers.h" #include "cublas_wrappers.h"
#include "kernels.h" #include "kernels.h"
template <typename T> template <typename T> class FeedForward {
class FeedForward { public:
public:
struct Config { struct Config {
int outputSize; int outputSize;
int inputSize; int inputSize;
std::array<int, 3> gemm_algos; std::array<int, 3> gemm_algos;
Config(int outputs, int inputs) Config(int outputs, int inputs)
: outputSize(outputs), : outputSize(outputs), inputSize(inputs),
inputSize(inputs),
gemm_algos(std::array<int, 3>({99, 99, 99})) {} gemm_algos(std::array<int, 3>({99, 99, 99})) {}
}; };
...@@ -63,6 +61,6 @@ class FeedForward { ...@@ -63,6 +61,6 @@ class FeedForward {
config_.inputSize = inputSize; config_.inputSize = inputSize;
} }
private: private:
Config config_; Config config_;
}; };
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment