cavis

History

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

cudnn

libnd4j polishing (#273 )

2020-03-02 12:49:41 +03:00

mkldnn

Softmax BP mkldnn implementation (#301 )

2020-03-12 18:25:29 +03:00

README.md

Platform helpers (#8216 )

2019-09-11 21:50:28 +03:00

README.md

This folder contains platform-specific optimized implementations for custom operations