cavis/libnd4j/include/ops/declarable/platform/mkldnn
raver119 7a2ac800dd
Nullify (#304)
* initial commit

Signed-off-by: raver119 <raver119@gmail.com>

* bunch of tweaks

Signed-off-by: raver119 <raver119@gmail.com>

* hamming distance nullification

Signed-off-by: raver119 <raver119@gmail.com>

* Add output array value assignment for testing/debugging

Signed-off-by: Alex Black <blacka101@gmail.com>

* don't assign empty arrays

Signed-off-by: raver119 <raver119@gmail.com>

* conv2d/conv3d/depthwise2d nullified

Signed-off-by: raver119 <raver119@gmail.com>

* conv2d/conv3d/depthwise2d nullified

Signed-off-by: raver119 <raver119@gmail.com>

* conv2d/conv3d/depthwise2d nullified

Signed-off-by: raver119 <raver119@gmail.com>

* few more fixes

Signed-off-by: raver119 <raver119@gmail.com>

* im2col

Signed-off-by: raver119 <raver119@gmail.com>

* pooling?

Signed-off-by: raver119 <raver119@gmail.com>

* more nullified

Signed-off-by: raver119 <raver119@gmail.com>

* ismax nullified

Signed-off-by: raver119 <raver119@gmail.com>

* rollback ismax nullification

Signed-off-by: raver119 <raver119@gmail.com>

* synchronized cublas handle use on per-device basis

Signed-off-by: raver119 <raver119@gmail.com>

* hiding method from jcpp

Signed-off-by: raver119 <raver119@gmail.com>

* get rid of test assigns in DeclarableOp

Signed-off-by: raver119 <raver119@gmail.com>

* get rid of assigns

Signed-off-by: raver119 <raver119@gmail.com>

* proper deviceId is back

Signed-off-by: raver119 <raver119@gmail.com>

* include fixed

Signed-off-by: raver119 <raver119@gmail.com>

Co-authored-by: Alex Black <blacka101@gmail.com>
2020-03-20 08:49:28 +03:00
..
avgpooling2d.cpp Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
avgpooling3d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batchnorm.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
conv2d.cpp Nullify (#304) 2020-03-20 08:49:28 +03:00
conv3d.cpp Nullify (#304) 2020-03-20 08:49:28 +03:00
deconv2d.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
deconv2d_tf.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
deconv3d.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
depthwiseConv2d.cpp Nullify (#304) 2020-03-20 08:49:28 +03:00
lrn.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
matmul.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
maxpooling2d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
maxpooling3d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
mkldnnUtils.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
mkldnnUtils.h Tanh backpropagation mkldnn implementation (#308) 2020-03-13 19:01:00 +03:00
softmax.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
tanh.cpp Tanh backpropagation mkldnn implementation (#308) 2020-03-13 19:01:00 +03:00