cavis/libnd4j/include/ops
Yurii Shyrma 00fd50cee2
Shyrma softmax (#209)
* - provide new cuda kernel for softmax

Signed-off-by: Yurii <yurii@skymind.io>

* - further work on cuda kernel for softmax

Signed-off-by: Yurii <yurii@skymind.io>

* - correction cuda kernel for softmax

Signed-off-by: Yurii <yurii@skymind.io>
2019-08-30 20:31:05 +03:00
..
declarable Shyrma softmax (#209) 2019-08-30 20:31:05 +03:00
impl [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
BroadcastBoolOpsTuple.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
BroadcastIntOpsTuple.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
BroadcastOpsTuple.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
InputType.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
aggregate_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
gemm.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
meta_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
ops.h [WIP] Int broadcastables (#195) 2019-08-30 10:12:40 +03:00
random_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
special_accumulation_ops.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00
special_ops.h [WIP] More CUDA fixes/updates (#62) 2019-07-20 23:07:42 +10:00
special_random_ops.h [WIP] More CUDA fixes/updates (#62) 2019-07-20 23:07:42 +10:00
specials.h [WIP] CUDA tweaks (#60) 2019-07-20 23:06:48 +10:00
specials_cuda.h Merge master to upstream (#7945) 2019-06-27 18:37:04 +03:00
specials_sparse.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +03:00