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

Include TF Import tests as forward pass tests in OpValidation (#53) 

* ignore multinomial

Signed-off-by: Ryan Nett <rnett@skymind.io>

* fix for @NonNull varargs

Signed-off-by: Ryan Nett <rnett@skymind.io>
master
Ryan Nett 2019-07-11 21:15:17 -07:00 committed by AlexDBlack
parent daf3950d8d
commit 62c6a73f9d
3 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -521,6 +521,12 @@ public class OpValidation {
currCount++;
opsWithTFMappingTFImportCounts.put(d.getClass(), currCount);
currCount = fwdPassCoverageCountPerClass.get(d.getClass());
if(currCount == null)
currCount = 0;
currCount++;
fwdPassCoverageCountPerClass.put(d.getClass(), currCount);
for(String s : tfNames){
currCount = tfMappedOpsImportTestCounts.get(s);
if(currCount == null)

2
nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/autodiff/opvalidation/ReductionOpValidation.java
View File

@ -827,7 +827,7 @@ public class ReductionOpValidation extends BaseOpValidation {
int[] d = dims.get(t);
for (int i = 0; i < 7; i++) {
int[] dim = d.length == 0 ? null : d;
int[] dim = d.length == 0 ? new int[0] : d;
SameDiff sd = SameDiff.create();
SDVariable s = sd.var("in", in);

4
nd4j/nd4j-backends/nd4j-tests/src/test/java/org/nd4j/imports/TFGraphs/TFGraphTestAllSameDiff.java
View File

@ -109,9 +109,9 @@ public class TFGraphTestAllSameDiff { //Note: Can't extend BaseNd4jTest here a
//2019/06/22 - Known issue: https://github.com/eclipse/deeplearning4j/issues/7935
"fake_quant/min_max_vars/.*",
"fake_quant/min_max_args/.*",
//2019/07/09 - Need "Multinomial" op - https://github.com/eclipse/deeplearning4j/issues/7913
"multinormal/.*"
"multinomial/.*"
};
@BeforeClass