| 1 | #include "common.cuh" |
|---|---|
| 2 | |
| 3 | #define CUDA_IM2COL_BLOCK_SIZE 256 |
| 4 | |
| 5 | void ggml_cuda_op_im2col(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 | void ggml_cuda_op_im2col_3d(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 7 |
| 1 | #include "common.cuh" |
|---|---|
| 2 | |
| 3 | #define CUDA_IM2COL_BLOCK_SIZE 256 |
| 4 | |
| 5 | void ggml_cuda_op_im2col(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 6 | void ggml_cuda_op_im2col_3d(ggml_backend_cuda_context & ctx, ggml_tensor * dst); |
| 7 |