Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test fix (#179)

Browse Source 
Signed-off-by: Alex Black <blacka101@gmail.com>
This commit is contained in:
Alex Black 2019-08-27 20:43:36 +10:00 committed by GitHub
parent a49f7c908b
commit dff599aa8f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/LayerOpValidation.java
View File

@ -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);
}