Class LibMatrixDNNIm2Col
java.lang.Object
org.apache.sysds.runtime.matrix.data.LibMatrixDNNIm2Col
This class contains the different implementation of im2col operation
-
Constructor Summary
Constructors -
Method Summary
Modifier and TypeMethodDescriptionstatic voidcol2imOverSingleImage(int outputN, MatrixBlock input, DnnParameters params) static voidim2col(MatrixBlock in, MatrixBlock out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) static voidim2col(MatrixBlock in, MatrixBlock out, int r, DnnParameters params, boolean trans) static voidim2colDense(double[] in, double[] out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) static voidim2colDenseStride1Pad0(double[] in, double[] out, int ai, int C, int R, int S, int H, int W, int P, int Q) static voidim2colSparse(MatrixBlock in, MatrixBlock out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) static void
-
Constructor Details
-
LibMatrixDNNIm2Col
public LibMatrixDNNIm2Col()
-
-
Method Details
-
im2col
public static void im2col(MatrixBlock in, MatrixBlock out, int r, DnnParameters params, boolean trans) -
im2col
public static void im2col(MatrixBlock in, MatrixBlock out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) -
im2colDenseStride1Pad0
public static void im2colDenseStride1Pad0(double[] in, double[] out, int ai, int C, int R, int S, int H, int W, int P, int Q) -
im2colDense
public static void im2colDense(double[] in, double[] out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) -
im2colSparse
public static void im2colSparse(MatrixBlock in, MatrixBlock out, int r, int C, int R, int S, int H, int W, int P, int Q, int stride_h, int stride_w, int pad_h, int pad_w, boolean trans) -
col2imOverSingleImage
-
preallocateSparseOutput
-