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, lstm, lstm_backward- See Also:
- Serialized Form
-
-
Field Summary
Fields Modifier and Type Field Description MatrixBlock
bias
int
C
int
D
boolean
enableNative
int[]
end_indexes_h
int[]
end_indexes_w
int
H
MatrixBlock
input1
MatrixBlock
input2
MatrixBlock
input3
MatrixBlock
input4
MatrixBlock
input5
MatrixBlock
input6
MatrixBlock
input7
MatrixBlock
input8
MatrixBlock
input9
int
K
int
M
double
minValForMaxPoolOperations
int
N
int
numThreads
MatrixBlock
output
MatrixBlock
output2
MatrixBlock
output3
MatrixBlock
output4
MatrixBlock
output5
int
P
int
pad_h
int
pad_w
int
Q
int
R
boolean
return_sequences
int
S
int[]
start_indexes_h
int[]
start_indexes_w
int
stride_h
int
stride_w
int
T
int
W
-
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(int N, int D, int T, int M, MatrixBlock x, MatrixBlock w, MatrixBlock bias, MatrixBlock out0, MatrixBlock c0, boolean return_sequences, int numThreads)
DnnParameters(int n, int d, int t, int m, MatrixBlock x, MatrixBlock w, MatrixBlock bias, MatrixBlock out0, MatrixBlock c0, MatrixBlock cache_out, MatrixBlock cache_c, MatrixBlock cache_ifog, boolean return_sequences, MatrixBlock dout, MatrixBlock dc, MatrixBlock dx, MatrixBlock dw, MatrixBlock db, MatrixBlock dout0, MatrixBlock dc0, 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 boolean
compare(DnnParameters that)
boolean
isAllOnes(Integer... params)
boolean
isOutputThreadSafe()
boolean
isStride1Pad0()
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)
String
toString()
-
-
-
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
-
D
public int D
-
T
public int T
-
M
public int M
-
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
-
return_sequences
public boolean return_sequences
-
input1
public MatrixBlock input1
-
input2
public MatrixBlock input2
-
output
public MatrixBlock output
-
input3
public MatrixBlock input3
-
input4
public MatrixBlock input4
-
input5
public MatrixBlock input5
-
input6
public MatrixBlock input6
-
input7
public MatrixBlock input7
-
input8
public MatrixBlock input8
-
input9
public MatrixBlock input9
-
output2
public MatrixBlock output2
-
output3
public MatrixBlock output3
-
output4
public MatrixBlock output4
-
output5
public MatrixBlock output5
-
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)
-
DnnParameters
public DnnParameters(int N, int D, int T, int M, MatrixBlock x, MatrixBlock w, MatrixBlock bias, MatrixBlock out0, MatrixBlock c0, boolean return_sequences, int numThreads)
-
DnnParameters
public DnnParameters(int n, int d, int t, int m, MatrixBlock x, MatrixBlock w, MatrixBlock bias, MatrixBlock out0, MatrixBlock c0, MatrixBlock cache_out, MatrixBlock cache_c, MatrixBlock cache_ifog, boolean return_sequences, MatrixBlock dout, MatrixBlock dc, MatrixBlock dx, MatrixBlock dw, MatrixBlock db, MatrixBlock dout0, MatrixBlock dc0, 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)
-
-