Brutex/cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
cavis/libnd4j/include/ops
History
raver119 f8364997c0
[WIP] maxpool2d_bp fix (#160) 
* one test for maxpool2d_bp

Signed-off-by: raver119 <raver119@gmail.com>

* - maxpool2d_bp cuda fix for NaNs
- streamSync after each custom op execution

Signed-off-by: raver119 <raver119@gmail.com>
2019-08-24 09:20:57 +03:00
..
declarable
[WIP] maxpool2d_bp fix (#160)
2019-08-24 09:20:57 +03:00
impl
int -> long (#108)
2019-08-10 09:14:18 +03:00
aggregate_ops.h
Eclipse Migration Initial Commit
2019-06-06 15:21:15 +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
gemm.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
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_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
specials.h
[WIP] CUDA tweaks (#60)
2019-07-20 23:06:48 +10:00