cavis/libnd4j/include/ops/declarable/platform/mkldnn
Yurii Shyrma 76f3553679
Shyrma merge max ind (#443)
* - provide correct possible output types in mergeMaxIndex op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - cleaning up the unneeded backprop arg in reverse_bp op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - improve clipByNorm both ff and bp

Signed-off-by: Yurii <iuriish@yahoo.com>

* - implementation and testing clipByAvgNorm_bp op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - pass biases in any way in dnnl lstm op, they are zeros when user doesn't provide them to us

Signed-off-by: Yurii <iuriish@yahoo.com>

* - start working on mkldnn concat op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - further work on mkldnn concat

Signed-off-by: Yurii <iuriish@yahoo.com>

* missing declaration fix

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

* - polishing mkl ops

Signed-off-by: Yurii <iuriish@yahoo.com>

* - testing and fixing bugs in mkl concat op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - fix linkage error for windows cuda build

Signed-off-by: Yurii <iuriish@yahoo.com>

* - further conflicts resolving with master

Signed-off-by: Yurii <iuriish@yahoo.com>

* - fix format tags in mkldnn matmul op

Signed-off-by: Yurii <iuriish@yahoo.com>

* - provide additional type cast in clip.cu

Signed-off-by: Yurii <iuriish@yahoo.com>

* - finally bug in mkldnn tanh_bp was caught

Co-authored-by: raver119@gmail.com <raver119@gmail.com>
2020-05-12 07:47:09 +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 Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
concat.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
conv2d.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
conv3d.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
deconv2d.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
deconv2d_tf.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
deconv3d.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
depthwiseConv2d.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
lrn.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
matmul.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +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 merge max ind (#443) 2020-05-12 07:47:09 +03:00
mkldnnUtils.h Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
softmax.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
tanh.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00
xw_plus_b.cpp Shyrma merge max ind (#443) 2020-05-12 07:47:09 +03:00