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

Another small fix (#175) 

* Layer norm 4d case fixes

Signed-off-by: AlexDBlack <blacka101@gmail.com>

* Small fix

Signed-off-by: AlexDBlack <blacka101@gmail.com>
master
Alex Black 2019-08-27 19:10:31 +10:00 committed by GitHub
parent dce4751fc1
commit 5cfbeb64ac
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/autodiff/validation/OpValidation.java
View File

@ -341,8 +341,8 @@ public class OpValidation {
//Finally: check execution/output
Map<String,INDArray> outOrig = original.execAll(tc.placeholderValues());
Map<String,INDArray> outDe = deserialized.execAll(tc.placeholderValues());
Map<String,INDArray> outOrig = original.outputAll(tc.placeholderValues());
Map<String,INDArray> outDe = deserialized.outputAll(tc.placeholderValues());
Preconditions.checkState(outOrig.keySet().equals(outDe.keySet()), "Keysets for execution after deserialization does not match key set for original model");
for(String s : outOrig.keySet()){

1
nd4j/nd4j-backends/nd4j-api-parent/nd4j-api/src/main/java/org/nd4j/imports/converters/DifferentialFunctionClassHolder.java
View File

@ -64,6 +64,7 @@ public class DifferentialFunctionClassHolder {
add("outputVariables");
add("tArguments");
add("iArguments");
add("bArguments");
add("hash");
add("opName");
add("sameDiff");