diff --git a/mace/ops/batch_norm_test.cc b/mace/ops/batch_norm_test.cc index 7f1bb0379865e8bb68b0fb86ceabcbea2090a4e0..8bb742f1c562da065419161b2d5370d35098ed33 100644 --- a/mace/ops/batch_norm_test.cc +++ b/mace/ops/batch_norm_test.cc @@ -115,7 +115,7 @@ TEST_F(BatchNormOpTest, SimpleRandomOPENCL) { net.AddRandomInput("Scale", {channels}); net.AddRandomInput("Offset", {channels}); net.AddRandomInput("Mean", {channels}); - net.AddRandomInput("Var", {channels}, true); + net.AddRandomInput("Var", {channels}); // run cpu net.RunOp(); @@ -176,7 +176,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) { .Input("Offset") .Input("Mean") .Input("Var") - .AddFloatArg("epsilon", 1e-3) + .AddFloatArg("epsilon", 1e-1) .Output("Output") .Finalize(net.NewOperatorDef()); @@ -186,7 +186,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) { net.AddRandomInput("Scale", {channels}); net.AddRandomInput("Offset", {channels}); net.AddRandomInput("Mean", {channels}); - net.AddRandomInput("Var", {channels}, true); + net.AddRandomInput("Var", {channels}); // run cpu net.RunOp(); @@ -213,7 +213,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) { .Input("OffsetImage") .Input("MeanImage") .Input("VarImage") - .AddFloatArg("epsilon", 1e-3) + .AddFloatArg("epsilon", 1e-1) .Output("OutputImage") .AddIntArg("T", static_cast(DataType::DT_HALF)) .Finalize(net.NewOperatorDef()); @@ -258,7 +258,7 @@ TEST_F(BatchNormOpTest, ComplexRandomOPENCL) { net.AddRandomInput("Scale", {channels}); net.AddRandomInput("Offset", {channels}); net.AddRandomInput("Mean", {channels}); - net.AddRandomInput("Var", {channels}, true); + net.AddRandomInput("Var", {channels}); // run cpu net.RunOp(); @@ -319,7 +319,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) { .Input("Offset") .Input("Mean") .Input("Var") - .AddFloatArg("epsilon", 1e-3) + .AddFloatArg("epsilon", 1e-1) .Output("Output") .Finalize(net.NewOperatorDef()); @@ -329,7 +329,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) { net.AddRandomInput("Scale", {channels}); net.AddRandomInput("Offset", {channels}); net.AddRandomInput("Mean", {channels}); - net.AddRandomInput("Var", {channels}, true); + net.AddRandomInput("Var", {channels}); // run cpu net.RunOp(); @@ -356,7 +356,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) { .Input("OffsetImage") .Input("MeanImage") .Input("VarImage") - .AddFloatArg("epsilon", 1e-3) + .AddFloatArg("epsilon", 1e-1) .Output("OutputImage") .AddIntArg("T", static_cast(DataType::DT_HALF)) .Finalize(net.NewOperatorDef()); @@ -403,7 +403,7 @@ TEST_F(BatchNormOpTest, NEONTest) { net.AddRandomInput("Scale", {channels}); net.AddRandomInput("Offset", {channels}); net.AddRandomInput("Mean", {channels}); - net.AddRandomInput("Var", {channels}, true); + net.AddRandomInput("Var", {channels}); // run cpu net.RunOp();