Uses of Class org.apache.sysds.runtime.matrix.data.LibMatrixCuDNNPoolingDescriptors (Apache SystemDS 3.4.0 API)

Packages that use LibMatrixCuDNNPoolingDescriptors

Package

Description

org.apache.sysds.runtime.matrix.data

Uses of LibMatrixCuDNNPoolingDescriptors in org.apache.sysds.runtime.matrix.data

Methods in org.apache.sysds.runtime.matrix.data that return LibMatrixCuDNNPoolingDescriptors

Modifier and Type

Method

Description

static LibMatrixCuDNNPoolingDescriptors

LibMatrixCuDNNPoolingDescriptors.cudnnPoolingBackwardDescriptors(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, LibMatrixDNN.PoolingType poolingType)

Get descriptors for maxpooling backward operation

static LibMatrixCuDNNPoolingDescriptors

LibMatrixCuDNNPoolingDescriptors.cudnnPoolingDescriptors(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, LibMatrixDNN.PoolingType poolingType)

Get descriptors for maxpooling operation