cavis/libnd4j/include/ops/declarable/platform/mkldnn
Yurii Shyrma e700b59f80
Shyrma weights format (#329)
* - start to introduce additional weights formats into conv2d ops

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

* - provide weights format variety in backprop conv2d and deconv2d ops, testing and fixing bugs

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

* - forgot to recover kernels sizes in deconv2d_bp test

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

* - built in weights format in depthwise conv 2d op

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

* - provide new weights formats in mkl dnn conv ops

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

* - provide new weights formats in cuda conv helpers

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

* - working with new weights format in cudnn conv api

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

* - take into account order of arrays in cudnn tensor descriptions

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

* - provide new weights formats in cpu conv3d (ff/bp)

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

* - provide new weights formats in cpu deconv3d (ff/bp)

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

* - provide new weights formats in conv3d ops (ff/bp) based on mkl api

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

* - provide new weights formats in conv3d ops (ff/bp) based on cudnn api

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

* - resolve conflicts 2

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

Co-authored-by: raver119 <raver119@gmail.com>
2020-03-20 12:11:27 +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 weights format (#329) 2020-03-20 12:11:27 +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 Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
matmul.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +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 Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
softmax.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00
tanh.cpp Shyrma weights format (#329) 2020-03-20 12:11:27 +03:00