cavis/libnd4j/include/ops/declarable/platform/mkldnn
Oleh e7a995e959
Tanh backpropagation mkldnn implementation (#308)
* libnd4j first step of tanh_bp operation implementation on mkldnn

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j optimize several places and added test case for tanh_bp

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j minor corrections and renaming, added one more test case

Signed-off-by: Oleg <oleg.semeniv@gmail.com>

* libnd4j missed mkldnn data format definition

Signed-off-by: Oleg <oleg.semeniv@gmail.com>
2020-03-13 19:01:00 +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 Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
conv3d.cpp Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +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 Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +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