Laurent Mazare 98d1242b8f im2col based conv2d (#802) ...

* im2col implementation for conv2d.

* Fix for the im2col implementation to match the current conv2d.

* Small optimization.

* Add a cuda kernel.

* Handle arbitrary layouts.

* Im2Col cuda code.

2023-09-10 21:02:42 +01:00

..

affine.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

binary_op_macros.cuh

Add some cmp tests. (#233)

2023-07-24 16:53:45 +01:00

binary.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

cast.cu

Let's keep the dirty code on its own.

2023-08-25 12:01:58 +00:00

compatibility.cuh

Compat windows.

2023-08-10 17:46:47 +02:00

conv.cu

im2col based conv2d (#802)

2023-09-10 21:02:42 +01:00

cuda_utils.cuh

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

fill.cu

Refactor the hierarchy.

2023-06-27 11:57:27 +02:00

indexing.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

lib.rs

Cuda kernels for IndexAdd/ScatterAdd. (#236)

2023-07-24 21:53:08 +01:00

reduce.cu

Add a dedicated cuda kernel for softmax. (#746)

2023-09-05 17:53:20 +02:00

ternary.cu

Add support for i64 (#563)

2023-08-23 10:42:19 +01:00

unary.cu

Add tanh. (#675)

2023-08-30 13:54:50 +01:00