2a488efb1b
* First steps for DL4J NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Conv2d NHWC forward pass works Signed-off-by: Alex Black <blacka101@gmail.com> * Conv2d NHWC backprop Signed-off-by: Alex Black <blacka101@gmail.com> * Conv2d backprop + fixes; subsampling fwd/bwd; improve tests Signed-off-by: Alex Black <blacka101@gmail.com> * Zero padding layer NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Cropping2D NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Deconv2d NHWC + clean up NHWC test framework code duplication Signed-off-by: Alex Black <blacka101@gmail.com> * CnnLossLayer NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Upsampling and batchnorm NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Space to depth Signed-off-by: Alex Black <blacka101@gmail.com> * Depthwise pt1 Signed-off-by: Alex Black <blacka101@gmail.com> * Depthwise pt2 and LRN Signed-off-by: Alex Black <blacka101@gmail.com> * SpaceToBatch Signed-off-by: Alex Black <blacka101@gmail.com> * LocallyConnected2D Signed-off-by: Alex Black <blacka101@gmail.com> * Fix depthwise nhwc support Signed-off-by: Alex Black <blacka101@gmail.com> * Upsampling NHWC - workaround for #8857 Signed-off-by: Alex Black <blacka101@gmail.com> * Workaround for #8859 - SpaceToDepth Signed-off-by: Alex Black <blacka101@gmail.com> * Batch normalization workaround - #8860 Signed-off-by: Alex Black <blacka101@gmail.com> * cuDNN fixes Signed-off-by: Alex Black <blacka101@gmail.com> * Switch cudnn conv2d to permute based impl due to 'true' NHWC not working Signed-off-by: Alex Black <blacka101@gmail.com> * cuDNN subsampling helper NHWC fix Signed-off-by: Alex Black <blacka101@gmail.com> * Upsampling/batchnorm fixes Signed-off-by: Alex Black <blacka101@gmail.com> * Small fixes Signed-off-by: Alex Black <blacka101@gmail.com> * CNN2D NHWC gradient checks (make CNNGradientCheckTest parameterized) Signed-off-by: Alex Black <blacka101@gmail.com> * Gradient checks, SConv2d, bunch of fixes Signed-off-by: Alex Black <blacka101@gmail.com> * Small fixes Signed-off-by: Alex Black <blacka101@gmail.com> * Global pooling NHWC support Signed-off-by: Alex Black <blacka101@gmail.com> * Also test both float and double for cuDNN NHWC tests Signed-off-by: Alex Black <blacka101@gmail.com> * Javadoc Signed-off-by: Alex Black <blacka101@gmail.com> * Ignore failing keras import test until next PR Signed-off-by: Alex Black <blacka101@gmail.com> |
||
|---|---|---|
| .. | ||
| src | ||
| pom.xml | ||