mirror of https://github.com/hpcaitech/ColossalAI
fix format (#574)
parent
dfe423ae42
commit
174b9c1d85
|
@ -10,9 +10,8 @@
|
|||
|
||||
using namespace std;
|
||||
|
||||
template <typename T>
|
||||
class Normalize_Layer {
|
||||
public:
|
||||
template <typename T> class Normalize_Layer {
|
||||
public:
|
||||
struct Config {
|
||||
uint32_t hidden_dim;
|
||||
bool use_mean;
|
||||
|
@ -58,7 +57,7 @@ class Normalize_Layer {
|
|||
|
||||
inline bool use_mean() const { return config_.use_mean; }
|
||||
|
||||
private:
|
||||
private:
|
||||
Config config_;
|
||||
T *vars_;
|
||||
T *means_;
|
||||
|
|
Loading…
Reference in New Issue