Class DnnParameters
- java.lang.Object
-
- org.apache.sysds.runtime.matrix.data.DnnParameters
-
- All Implemented Interfaces:
Serializable
public class DnnParameters extends Object implements Serializable
This class is container that stores parameters required for executing following operations: conv2d, conv2d_backward_data, conv2d_backward_filter, maxpooling, maxpooling_backward- See Also:
- Serialized Form
-
-
Field Summary
Fields Modifier and Type Field Description MatrixBlockbiasintCbooleanenableNativeint[]end_indexes_hint[]end_indexes_wintHMatrixBlockinput1MatrixBlockinput2intKdoubleminValForMaxPoolOperationsintNintnumThreadsMatrixBlockoutputintPintpad_hintpad_wintQintRintSint[]start_indexes_hint[]start_indexes_wintstride_hintstride_wintW
-
Constructor Summary
Constructors Constructor Description DnnParameters(int N, int C, int H, int W, int K, int R, int S, int stride_h, int stride_w, int pad_h, int pad_w, int numThreads)DnnParameters(long N, long C, long H, long W, long K, long R, long S, long stride_h, long stride_w, long pad_h, long pad_w, int numThreads)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description booleancompare(DnnParameters that)booleanisAllOnes(Integer... params)booleanisOutputThreadSafe()booleanisStride1Pad0()voidsetIfUnknown(Hop N, Hop C, Hop H, Hop W, Hop K, Hop R, Hop S, Hop stride_h, Hop stride_w, Hop pad_h, Hop pad_w, int numThreads)StringtoString()
-
-
-
Field Detail
-
N
public int N
-
C
public int C
-
H
public int H
-
W
public int W
-
K
public int K
-
R
public int R
-
S
public int S
-
P
public int P
-
Q
public int Q
-
stride_h
public int stride_h
-
stride_w
public int stride_w
-
pad_h
public int pad_h
-
pad_w
public int pad_w
-
numThreads
public int numThreads
-
enableNative
public boolean enableNative
-
input1
public MatrixBlock input1
-
input2
public MatrixBlock input2
-
output
public MatrixBlock output
-
bias
public MatrixBlock bias
-
start_indexes_h
public int[] start_indexes_h
-
end_indexes_h
public int[] end_indexes_h
-
start_indexes_w
public int[] start_indexes_w
-
end_indexes_w
public int[] end_indexes_w
-
minValForMaxPoolOperations
public double minValForMaxPoolOperations
-
-
Constructor Detail
-
DnnParameters
public DnnParameters(long N, long C, long H, long W, long K, long R, long S, long stride_h, long stride_w, long pad_h, long pad_w, int numThreads)
-
DnnParameters
public DnnParameters(int N, int C, int H, int W, int K, int R, int S, int stride_h, int stride_w, int pad_h, int pad_w, int numThreads)
-
-
Method Detail
-
compare
public boolean compare(DnnParameters that)
-
setIfUnknown
public void setIfUnknown(Hop N, Hop C, Hop H, Hop W, Hop K, Hop R, Hop S, Hop stride_h, Hop stride_w, Hop pad_h, Hop pad_w, int numThreads)
-
isOutputThreadSafe
public boolean isOutputThreadSafe()
-
isStride1Pad0
public boolean isStride1Pad0()
-
isAllOnes
public boolean isAllOnes(Integer... params)
-
-