@@ -17,7 +17,7 @@ TEST(DropOutLayer, IncompatibleInput) {
1717}
1818
1919TEST (DropOutLayer, dropoutlayer_int) {
20- DropOutLayer layer (1 );
20+ DropOutLayer layer (1 , true );
2121 Shape sh ({2 , 2 });
2222 Tensor input = make_tensor<int >({1 , -1 , 2 , -2 }, sh);
2323 Tensor output;
@@ -47,7 +47,7 @@ TEST(DropOutLayer, dropoutlayer_float) {
4747}
4848
4949TEST (DropOutLayer, dropoutlayer_float_50proc) {
50- DropOutLayer layer (0.5 );
50+ DropOutLayer layer (0.5 , true );
5151 Shape sh ({10 , 10 });
5252 std::vector<float > a (100 , static_cast <float >(0.01 ));
5353 Tensor input = make_tensor<float >(a, sh);
@@ -60,7 +60,7 @@ TEST(DropOutLayer, dropoutlayer_float_50proc) {
6060}
6161
6262TEST (DropOutLayer, dropoutlayer_float_30proc) {
63- DropOutLayer layer (0.3 );
63+ DropOutLayer layer (0.3 , true );
6464 Shape sh ({10 , 10 });
6565 std::vector<float > a (100 , static_cast <float >(0.01 ));
6666 Tensor input = make_tensor<float >(a, sh);
@@ -73,7 +73,7 @@ TEST(DropOutLayer, dropoutlayer_float_30proc) {
7373}
7474
7575TEST (DropOutLayer, dropoutlayer_float_70proc) {
76- DropOutLayer layer (0.7 );
76+ DropOutLayer layer (0.7 , true );
7777 Shape sh ({10 , 10 });
7878 std::vector<float > a (100 , static_cast <float >(0.01 ));
7979 Tensor input = make_tensor<float >(a, sh);
0 commit comments