cavis/libnd4j/include/ops
Alex Black e855e47f73
More fixes (#148)
* Small batch norm fix (cuda/no-mkldnn)

Signed-off-by: Alex Black <blacka101@gmail.com>

* Dropout fix for RnnOutputLayer

Signed-off-by: Alex Black <blacka101@gmail.com>

* Allow block size < 2 in batch_to_space_nd and space_to_batch_nd for import, in spite of what TF docs say

Signed-off-by: AlexDBlack <blacka101@gmail.com>
2019-08-22 19:55:27 +10:00
..
declarable More fixes (#148) 2019-08-22 19:55:27 +10:00
impl int -> long (#108) 2019-08-10 09:14:18 +03:00
BroadcastBoolOpsTuple.h Eclipse Migration Initial Commit 2019-06-06 15:21:15 +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 Merge master to upstream (#7945) 2019-06-27 18:37:04 +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