org.apache.sysml.runtime.matrix.data

Class LibMatrixNative

java.lang.Object

org.apache.sysml.runtime.matrix.data.LibMatrixNative

public class LibMatrixNative
extends Object

Constructor Summary

Constructors

Constructor and Description
LibMatrixNative()

Method Summary

Modifier and Type	Method and Description
static void	conv2d(MatrixBlock input, MatrixBlock filter, MatrixBlock outputBlock, ConvolutionParameters params) This method performs convolution (i.e.
static void	conv2dBackwardData(MatrixBlock filter, MatrixBlock dout, MatrixBlock outputBlock, ConvolutionParameters params) This method computes the backpropogation errors for previous layer of convolution operation
static void	conv2dBackwardFilter(MatrixBlock input, MatrixBlock dout, MatrixBlock outputBlock, ConvolutionParameters params) This method computes the backpropogation errors for filter of convolution operation
static void	matrixMult(MatrixBlock m1, MatrixBlock m2, MatrixBlock ret, int k) Performs matrix multiplication using native library if BLAS is available or else falls back to Java BLAS.
static void	matrixMult(MatrixBlock m1, MatrixBlock m2, MatrixBlock ret, int k, boolean examSparsity)

Methods inherited from class java.lang.Object

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

Constructor Detail

LibMatrixNative

public LibMatrixNative()

Method Detail

matrixMult

public static void matrixMult(MatrixBlock m1,
                              MatrixBlock m2,
                              MatrixBlock ret,
                              int k)
                       throws DMLRuntimeException

Performs matrix multiplication using native library if BLAS is available or else falls back to Java BLAS.

Parameters:

m1 - lhs matrix block

m2 - rhs matrix block

ret - output matrix block

k - number of threads

Throws:

DMLRuntimeException - if error occurs

matrixMult

public static void matrixMult(MatrixBlock m1,
                              MatrixBlock m2,
                              MatrixBlock ret,
                              int k,
                              boolean examSparsity)
                       throws DMLRuntimeException

Throws:

DMLRuntimeException

conv2d

public static void conv2d(MatrixBlock input,
                          MatrixBlock filter,
                          MatrixBlock outputBlock,
                          ConvolutionParameters params)
                   throws DMLRuntimeException

This method performs convolution (i.e. cross-correlation) operation on input

Parameters:

input - input batch

filter - filter

outputBlock - output of convolution

params - convolution parameters

Throws:

DMLRuntimeException - if DMLRuntimeException occurs

conv2dBackwardFilter

public static void conv2dBackwardFilter(MatrixBlock input,
                                        MatrixBlock dout,
                                        MatrixBlock outputBlock,
                                        ConvolutionParameters params)
                                 throws DMLRuntimeException

This method computes the backpropogation errors for filter of convolution operation

Parameters:

input - input image

dout - errors from next layer

outputBlock - output errors

params - convolution parameters

Throws:

DMLRuntimeException - if DMLRuntimeException occurs

conv2dBackwardData

public static void conv2dBackwardData(MatrixBlock filter,
                                      MatrixBlock dout,
                                      MatrixBlock outputBlock,
                                      ConvolutionParameters params)
                               throws DMLRuntimeException

This method computes the backpropogation errors for previous layer of convolution operation

Parameters:

filter - filter used in conv2d

dout - errors from next layer

outputBlock - output errors

params - convolution parameters

Throws:

DMLRuntimeException - if DMLRuntimeException occurs