public class LibMatrixNative extends Object
Constructor and Description |
---|
LibMatrixNative() |
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) |
public static void matrixMult(MatrixBlock m1, MatrixBlock m2, MatrixBlock ret, int k) throws DMLRuntimeException
m1
- lhs matrix blockm2
- rhs matrix blockret
- output matrix blockk
- number of threadsDMLRuntimeException
- if error occurspublic static void matrixMult(MatrixBlock m1, MatrixBlock m2, MatrixBlock ret, int k, boolean examSparsity) throws DMLRuntimeException
DMLRuntimeException
public static void conv2d(MatrixBlock input, MatrixBlock filter, MatrixBlock outputBlock, ConvolutionParameters params) throws DMLRuntimeException
input
- input batchfilter
- filteroutputBlock
- output of convolutionparams
- convolution parametersDMLRuntimeException
- if DMLRuntimeException occurspublic static void conv2dBackwardFilter(MatrixBlock input, MatrixBlock dout, MatrixBlock outputBlock, ConvolutionParameters params) throws DMLRuntimeException
input
- input imagedout
- errors from next layeroutputBlock
- output errorsparams
- convolution parametersDMLRuntimeException
- if DMLRuntimeException occurspublic static void conv2dBackwardData(MatrixBlock filter, MatrixBlock dout, MatrixBlock outputBlock, ConvolutionParameters params) throws DMLRuntimeException
filter
- filter used in conv2ddout
- errors from next layeroutputBlock
- output errorsparams
- convolution parametersDMLRuntimeException
- if DMLRuntimeException occursCopyright © 2017 The Apache Software Foundation. All rights reserved.