| Package | Description | 
|---|---|
| org.apache.sysds.runtime.matrix.data | 
| Modifier and Type | Method and Description | 
|---|---|
| static LibMatrixCuDNNConvolutionAlgorithm | LibMatrixCuDNNConvolutionAlgorithm. cudnnGetConvolutionBackwardDataAlgorithm(GPUContext gCtx,
                                        String instName,
                                        int N,
                                        int C,
                                        int H,
                                        int W,
                                        int K,
                                        int R,
                                        int S,
                                        int pad_h,
                                        int pad_w,
                                        int stride_h,
                                        int stride_w,
                                        int P,
                                        int Q,
                                        long workspaceLimit)Factory method to get the algorithm wrapper for convolution backward data | 
| static LibMatrixCuDNNConvolutionAlgorithm | LibMatrixCuDNNConvolutionAlgorithm. cudnnGetConvolutionBackwardFilterAlgorithm(GPUContext gCtx,
                                          String instName,
                                          int N,
                                          int C,
                                          int H,
                                          int W,
                                          int K,
                                          int R,
                                          int S,
                                          int pad_h,
                                          int pad_w,
                                          int stride_h,
                                          int stride_w,
                                          int P,
                                          int Q,
                                          long workspaceLimit)Factory method to get the algorithm wrapper for convolution backward filter | 
| static LibMatrixCuDNNConvolutionAlgorithm | LibMatrixCuDNNConvolutionAlgorithm. cudnnGetConvolutionForwardAlgorithm(GPUContext gCtx,
                                   String instName,
                                   int N,
                                   int C,
                                   int H,
                                   int W,
                                   int K,
                                   int R,
                                   int S,
                                   int pad_h,
                                   int pad_w,
                                   int stride_h,
                                   int stride_w,
                                   int P,
                                   int Q,
                                   long workspaceLimit)Factory method to get the algorithm wrapper for convolution forward | 
Copyright © 2021 The Apache Software Foundation. All rights reserved.