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

Fixes (#202) 

* Remove unnecessary dup in MKLDNNSubsamplingHelper

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

* gson dep fix

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

3
deeplearning4j/deeplearning4j-nn/src/main/java/org/deeplearning4j/nn/layers/mkldnn/MKLDNNSubsamplingHelper.java
View File

@ -69,9 +69,6 @@ public class MKLDNNSubsamplingHelper implements SubsamplingHelper {
pad = ConvolutionUtils.getSameModeTopLeftPadding(new int[]{(int)epsilon.size(2), (int)epsilon.size(3)}, new int[] {(int)input.size(2), (int)input.size(3)}, kernel, strides, dilation);
}
input = input.dup();
epsilon = epsilon.dup();
Pooling2DConfig conf = Pooling2DConfig.builder()
.isSameMode(convolutionMode == ConvolutionMode.Same)
.kH(kernel[0]).kW(kernel[1])

7
nd4j/nd4j-remote/nd4j-json-server/pom.xml
View File

@ -123,6 +123,13 @@
<artifactId>activation</artifactId>
<version>1.1</version>
</dependency>
<dependency>
<groupId>com.google.code.gson</groupId>
<artifactId>gson</artifactId>
<version>${gson.version}</version>
<scope>test</scope>
</dependency>
</dependencies>
<build>