cavis/libnd4j/include/ops/declarable/platform/mkldnn
Oleh 41bde8f885
Softmax BP mkldnn implementation (#301)
* libnd4j mkldnn softmax_bp operation implementation and integration, 2 tests added, need some refactoring and code clean up and more testing with different input shapes

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

* libnd4j softmax_bp update, code refactoring, etc

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

* libnd4j merge master, fixed typos, minor tweaks, code clean up

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

* libnd4j  integrate mkldnnUtils helpers in other mkldnn operations

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