com
.
intel
.
analytics
.
bigdl
.
nn
.
mkldnn
MklDnnMemory
Related Doc:
package mkldnn
object
MklDnnMemory
Linear Supertypes
AnyRef
,
Any
Ordering
Alphabetic
By Inheritance
Inherited
MklDnnMemory
AnyRef
Any
Hide All
Show All
Visibility
Public
All
Value Members
final
def
!=
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
final
def
##
()
:
Int
Definition Classes
AnyRef → Any
final
def
==
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
BatchNormBackwardDescInit
(
prop_kind:
Int
,
diffDstMemDesc:
Long
,
srcMemDesc:
Long
,
epsilon:
Float
,
flags:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
BatchNormForwardDescInit
(
propKind:
Int
,
srcMemDesc:
Long
,
epsilon:
Float
,
flags:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ConcatPrimitiveDescCreate
(
output_desc:
Long
,
n:
Int
,
concat_dimension:
Int
,
input_pds:
Array
[
Long
]
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ConvBackwardDataDescInit
(
alg_kind:
Int
,
diff_src_desc:
Long
,
weights_desc:
Long
,
diff_dst_desc:
Long
,
strides:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ConvBackwardWeightsDescInit
(
alg_kind:
Int
,
src_desc:
Long
,
diff_weights_desc:
Long
,
diff_bias_desc:
Long
,
diff_dst_desc:
Long
,
strides:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ConvForwardDescInit
(
prop_kind:
Int
,
alg_kind:
Int
,
src_desc:
Long
,
weights_desc:
Long
,
bias_desc:
Long
,
dst_desc:
Long
,
strides:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
CreateAttr
()
(
implicit
owner:
MemoryOwner
)
:
Long
def
CreatePostOps
()
(
implicit
owner:
MemoryOwner
)
:
Long
def
DilatedConvBackwardDataDescInit
(
alg_kind:
Int
,
diff_src_desc:
Long
,
weights_desc:
Long
,
diff_dst_desc:
Long
,
strides:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
dilates:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
DilatedConvBackwardWeightsDescInit
(
alg_kind:
Int
,
src_desc:
Long
,
diff_weights_desc:
Long
,
diff_bias_desc:
Long
,
diff_dst_desc:
Long
,
strides:
Array
[
Int
]
,
dilates:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
DilatedConvForwardDescInit
(
prop_kind:
Int
,
alg_kind:
Int
,
src_desc:
Long
,
weights_desc:
Long
,
bias_desc:
Long
,
dst_desc:
Long
,
strides:
Array
[
Int
]
,
dilates:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
EltwiseBackwardDescInit
(
algKind:
Int
,
diffDataDesc:
Long
,
dataDesc:
Long
,
alpha:
Float
,
beta:
Float
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
EltwiseForwardDescInit
(
propKind:
Int
,
algKind:
Int
,
srcDesc:
Long
,
alpha:
Float
,
beta:
Float
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
LRNBackwardDescInit
(
alg_kind:
Int
,
diff_data_desc:
Long
,
data_desc:
Long
,
local_size:
Int
,
alpha:
Float
,
beta:
Float
,
k:
Float
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
LRNForwardDescInit
(
prop_kind:
Int
,
alg_kind:
Int
,
data_desc:
Long
,
local_size:
Int
,
alpha:
Float
,
beta:
Float
,
k:
Float
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
LinearBackwardDataDescInit
(
diffSrcMemDesc:
Long
,
weightMemDesc:
Long
,
diffDstMemDesc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
LinearBackwardWeightsDescInit
(
srcMemDesc:
Long
,
diffWeightMemDesc:
Long
,
diffBiasMemDesc:
Long
,
diffDstMemDesc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
LinearForwardDescInit
(
propKind:
Int
,
srcMemDesc:
Long
,
weightMemDesc:
Long
,
biasMemDesc:
Long
,
dstMemDesc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
MemoryDescInit
(
ndims:
Int
,
dims:
Array
[
Int
]
,
dataType:
Int
,
dataFormat:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
MemoryPrimitiveDescCreate
(
desc:
Long
,
engine:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PoolingBackwardDescInit
(
alg_kind:
Int
,
diff_src_desc:
Long
,
diff_dst_desc:
Long
,
strides:
Array
[
Int
]
,
kernel:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PoolingForwardDescInit
(
prop_kind:
Int
,
alg_kind:
Int
,
src_desc:
Long
,
dst_desc:
Long
,
strides:
Array
[
Int
]
,
kernel:
Array
[
Int
]
,
padding_l:
Array
[
Int
]
,
padding_r:
Array
[
Int
]
,
padding_kind:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PrimitiveCreate0
(
desc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PrimitiveCreate2
(
desc:
Long
,
inputs:
Array
[
Long
]
,
indexes:
Array
[
Int
]
,
inputLen:
Int
,
outputs:
Array
[
Long
]
,
outputLen:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PrimitiveDescCreate
(
opDesc:
Long
,
engine:
Long
,
hingForwardPrimitiveDesc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
PrimitiveDescCreateV2
(
opDesc:
Long
,
attr:
Long
,
engine:
Long
,
hingForwardPrimitiveDesc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
RNNBackwardDescInit
(
prop_kind:
Int
,
rnn_cell_desc:
Long
,
direction:
Int
,
src_layer_desc:
Long
,
src_iter_desc:
Long
,
weights_layer_desc:
Long
,
weights_iter_desc:
Long
,
bias_desc:
Long
,
dst_layer_desc:
Long
,
dst_iter_desc:
Long
,
diff_src_layer_desc:
Long
,
diff_src_iter_desc:
Long
,
diff_weights_layer_desc:
Long
,
diff_weights_iter_desc:
Long
,
diff_bias_desc:
Long
,
diff_dst_layer_desc:
Long
,
diff_dst_iter_desc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
RNNCellDescInit
(
kind:
Int
,
f:
Int
,
flags:
Int
,
alpha:
Float
,
clipping:
Float
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
RNNForwardDescInit
(
prop_kind:
Int
,
rnn_cell_desc:
Long
,
direction:
Int
,
src_layer_desc:
Long
,
src_iter_desc:
Long
,
weights_layer_desc:
Long
,
weights_iter_desc:
Long
,
bias_desc:
Long
,
dst_layer_desc:
Long
,
dst_iter_desc:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ReorderPrimitiveDescCreate
(
input:
Long
,
output:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ReorderPrimitiveDescCreateV2
(
input:
Long
,
output:
Long
,
attr:
Long
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
SoftMaxBackwardDescInit
(
propKind:
Int
,
diffDesc:
Long
,
dstDesc:
Long
,
axis:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
SoftMaxForwardDescInit
(
prop_kind:
Int
,
dataDesc:
Long
,
axis:
Int
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
SumPrimitiveDescCreate
(
output_mem_desc:
Long
,
n:
Int
,
scales:
Array
[
Float
]
,
input_pds:
Array
[
Long
]
)
(
implicit
owner:
MemoryOwner
)
:
Long
def
ViewPrimitiveDescCreate
(
memory_primitive_desc:
Long
,
dims:
Array
[
Int
]
,
offsets:
Array
[
Int
]
)
(
implicit
owner:
MemoryOwner
)
:
Long
final
def
asInstanceOf
[
T0
]
:
T0
Definition Classes
Any
def
clone
()
:
AnyRef
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
eq
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
def
equals
(
arg0:
Any
)
:
Boolean
Definition Classes
AnyRef → Any
def
finalize
()
:
Unit
Attributes
protected[
java.lang
]
Definition Classes
AnyRef
Annotations
@throws
(
classOf[java.lang.Throwable]
)
final
def
getClass
()
:
Class
[_]
Definition Classes
AnyRef → Any
def
hashCode
()
:
Int
Definition Classes
AnyRef → Any
final
def
isInstanceOf
[
T0
]
:
Boolean
Definition Classes
Any
final
def
ne
(
arg0:
AnyRef
)
:
Boolean
Definition Classes
AnyRef
final
def
notify
()
:
Unit
Definition Classes
AnyRef
final
def
notifyAll
()
:
Unit
Definition Classes
AnyRef
final
def
synchronized
[
T0
]
(
arg0: ⇒
T0
)
:
T0
Definition Classes
AnyRef
def
toString
()
:
String
Definition Classes
AnyRef → Any
final
def
wait
()
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
,
arg1:
Int
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
final
def
wait
(
arg0:
Long
)
:
Unit
Definition Classes
AnyRef
Annotations
@throws
(
...
)
Inherited from
AnyRef
Inherited from
Any
Ungrouped
