Updated some parameters from double to float

Sable · Jun 14, 2019 · 6a58f17 · 6a58f17
1 parent d97ba78
commit 6a58f17
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/nn-builder/src/builtins/activation.h b/src/nn-builder/src/builtins/activation.h
@@ -11,9 +11,9 @@ struct ActivationFunction {
   wabt::Var derivative;
 };
 struct ActivationOptions {
-  double linear_slope = 1;
-  double leaky_relu_slope = 0.01;
-  double elu_slope = 0.01;
+  float linear_slope = 1;
+  float leaky_relu_slope = 0.01;
+  float elu_slope = 0.01;
 };
 class Activation : public Builtin {
 private: