Package | Description |
---|---|
org.apache.sysml.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 © 2018 The Apache Software Foundation. All rights reserved.