org.apache.sysml.runtime.matrix.data

Class LibMatrixDNNHelper

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixDNNHelper

public class LibMatrixDNNHelper
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
protected static class	LibMatrixDNNHelper.CellIndex3

Constructor Summary

Constructors

Constructor and Description
LibMatrixDNNHelper()

Method Summary

Modifier and Type	Method and Description
protected static LibMatrixDNNHelper.CellIndex3	computeTensorIndexes(int j, int H, int W)
protected static LibMatrixDNNHelper.CellIndex3	computeTensorIndexes(int j, int N, int M, LibMatrixDNNHelper.CellIndex3 ret) Computes tensor indexes from a linearized column index such that the column index is equal to ix1*NM + ix2*M + ix3
protected static void	singleThreadedMatMult(MatrixBlock m1, MatrixBlock m2, MatrixBlock ret, boolean recomputeNNZM1, boolean recomputeNNZM2, DnnParameters params)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LibMatrixDNNHelper

public LibMatrixDNNHelper()

Method Detail

computeTensorIndexes

protected static LibMatrixDNNHelper.CellIndex3 computeTensorIndexes(int j,
                                                                    int H,
                                                                    int W)

computeTensorIndexes

protected static LibMatrixDNNHelper.CellIndex3 computeTensorIndexes(int j,
                                                                    int N,
                                                                    int M,
                                                                    LibMatrixDNNHelper.CellIndex3 ret)

Computes tensor indexes from a linearized column index such that the column index is equal to ix1*NM + ix2*M + ix3

Parameters:

j - column index

N - second last dimension

M - last dimension

ret - output object for reuse

Returns:

tensor indexes

singleThreadedMatMult

protected static void singleThreadedMatMult(MatrixBlock m1,
                                            MatrixBlock m2,
                                            MatrixBlock ret,
                                            boolean recomputeNNZM1,
                                            boolean recomputeNNZM2,
                                            DnnParameters params)