cavis/libnd4j/include/ops
Oleh 41bde8f885
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
..
declarable Softmax BP mkldnn implementation (#301) 2020-03-12 18:25:29 +03:00
impl [WIP] Shyrma coords (#305) 2020-03-11 16:21:59 +03:00
BroadcastBoolOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BroadcastIntOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
BroadcastOpsTuple.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
InputType.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
gemm.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
meta_ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
random_ops.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00
special_random_ops.h Revert "OpenMP Threads execution (#297)" (#299) 2020-03-09 08:22:49 +03:00
specials.h profiling of stack and unstack ops (#261) 2020-03-03 07:32:37 +03:00
specials_cuda.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
specials_sparse.h libnd4j polishing (#273) 2020-03-02 12:49:41 +03:00