Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops/declarable/platform/mkldnn
History
Oleh 4d81af9fe9
Softmax operation implementation for mkldnn (#286) 
* libnd4j first step of softmax mkldnn implementation

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

* libnd4j raw implementation of mkldnn softmax

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

* libnd4j merge master and added softmax to MklDnnTests

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

* libnd4j some corrections for softmax mkldnn

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

* libnd4j merge branch, fixed problem with negative axis, fixed dnnl::memory::format_tag selection, test cases added

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

* libnd4j minor corrections to avoid risk connected with negative axis usage

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

* libnd4j fixed windows builds, added switcher to use mkldnn sofmax version only for 3D, 4D, 5D, 6D arrays

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

* libnd4j fixed dataType selection per request

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

* libnd4j fix for mac and windows builds

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

* libnd4j builds fix

Signed-off-by: Oleg <oleg.semeniv@gmail.com>
 		2020-03-04 19:36:42 +03:00
..
avgpooling2d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
avgpooling3d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
batchnorm.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
conv2d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
conv3d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
deconv2d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
deconv2d_tf.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
deconv3d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
depthwiseConv2d.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lrn.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
lstmLayer.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
matmul.cpp libnd4j polishing (#273) 2020-03-02 12:49:41 +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 libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
mkldnnUtils.h Softmax operation implementation for mkldnn (#286) 2020-03-04 19:36:42 +03:00
softmax.cpp Softmax operation implementation for mkldnn (#286) 2020-03-04 19:36:42 +03:00