cavis/libnd4j/include/ops/declarable/platform/mkldnn
raver119 c9d1454743
MKLDNN tweaks (#415)
* one simple test

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

* fix

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

* hmmmm...

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

* mkl matmul skip tweaks

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

* minor fix for MemoryTracker

* long shapes in matmul

* - 2 new tests for mkldnn tanh
- mkldnn isn't used for scalar tanh
2020-04-27 17:37:53 +03:00
..
avgpooling2d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
avgpooling3d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
batchnorm.cpp Fix for certain non-ews cases (#402) 2020-04-21 12:41:30 +03:00
conv2d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
conv3d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
deconv2d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
deconv2d_tf.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
deconv3d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
depthwiseConv2d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
lrn.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.cpp At cpp ops (#378) 2020-04-17 15:16:14 +10:00
matmul.cpp MKLDNN tweaks (#415) 2020-04-27 17:37:53 +03:00
maxpooling2d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
maxpooling3d.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
mkldnnUtils.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
mkldnnUtils.h xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +03:00
softmax.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
tanh.cpp MKLDNN tweaks (#415) 2020-04-27 17:37:53 +03:00
xw_plus_b.cpp xw_plus_b mkldnn implementation (#247) 2020-03-31 13:03:10 +03:00