cuda/cpu: add im2col_3d support

This commit is contained in:
leejet
2025-08-02 00:38:51 +08:00
parent 93c7e775b8
commit f7a12f9e69
8 changed files with 353 additions and 12 deletions

View File

@@ -511,6 +511,7 @@ extern "C" {
GGML_OP_CONV_TRANSPOSE_1D,
GGML_OP_IM2COL,
GGML_OP_IM2COL_BACK,
GGML_OP_IM2COL_3D,
GGML_OP_CONV_2D,
GGML_OP_CONV_3D,
GGML_OP_CONV_2D_DW,