Skip to content

Commit 50a7855

Browse files
committed
ggml: add support for 3D IM2COL operation in Hexagon capabilities
1 parent d07d8e6 commit 50a7855

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml/src/ggml-hexagon/ggml-hexagon.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -647,6 +647,7 @@ static constexpr const qnn_op_caps ggmlqnn_k_op_caps[] = {
647647
{false, GGML_OP_CONV_TRANSPOSE_1D, 0, nullptr},
648648
{false, GGML_OP_IM2COL, 0, nullptr},
649649
{false, GGML_OP_IM2COL_BACK, 0, nullptr},
650+
{false, GGML_OP_IM2COL_3D, 0, nullptr},
650651
{false, GGML_OP_CONV_2D, 0, nullptr},
651652
{false, GGML_OP_CONV_3D, 0, nullptr},
652653
{false, GGML_OP_CONV_2D_DW, 0, nullptr},
@@ -747,6 +748,7 @@ static constexpr const hexagon_op_caps ggmlhexagon_k_op_caps[] = {
747748
{false, GGML_OP_CONV_TRANSPOSE_1D, 0, nullptr, nullptr},
748749
{false, GGML_OP_IM2COL, 0, nullptr, nullptr},
749750
{false, GGML_OP_IM2COL_BACK, 0, nullptr, nullptr},
751+
{false, GGML_OP_IM2COL_3D, 0, nullptr, nullptr},
750752
{false, GGML_OP_CONV_2D, 0, nullptr, nullptr},
751753
{false, GGML_OP_CONV_3D, 0, nullptr, nullptr},
752754
{false, GGML_OP_CONV_2D_DW, 0, nullptr, nullptr},

0 commit comments

Comments
 (0)