| Package | Description |
|---|---|
| org.deeplearning4j.nn.layers.convolution |
| Modifier and Type | Method and Description |
|---|---|
static CudnnConvolutionHelper.CudnnForwardArgs |
CudnnConvolutionHelper.getCudnnForwardArgs(INDArray input,
int[] kernel,
int[] strides,
int[] padding,
int[] dilation,
org.deeplearning4j.nn.conf.ConvolutionMode convolutionMode,
org.deeplearning4j.nn.conf.layers.PoolingType poolingType) |
Copyright © 2018. All rights reserved.