From dff599aa8fbbaa45765bb4ea82d494082c6300ac Mon Sep 17 00:00:00 2001 From: Alex Black Date: Tue, 27 Aug 2019 20:43:36 +1000 Subject: [PATCH] Test fix (#179) Signed-off-by: Alex Black --- .../org/nd4j/autodiff/opvalidation/LayerOpValidation.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java index fde2170a6..760165b3b 100644 --- a/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java +++ b/nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java @@ -1178,7 +1178,7 @@ public class LayerOpValidation extends BaseOpValidation { final INDArray res = standardized.mulRowVector(gain).addRowVector(bias); final INDArray output = Nd4j.zerosLike(res); - Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, bias, output, 1)); + Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, bias, output, true, 1)); assertEquals(res, output); } @@ -1216,7 +1216,7 @@ public class LayerOpValidation extends BaseOpValidation { final INDArray res = standardized.mulRowVector(gain); final INDArray output = Nd4j.zerosLike(res); - Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, output, 1)); + Nd4j.getExecutioner().exec(new LayerNorm(standardized, gain, output, true, 1)); assertEquals(res, output); }