From ac4517b5c1c8e234317f5759c9d83bf0cbc7b163 Mon Sep 17 00:00:00 2001 From: Feiyue Chen Date: Fri, 16 Dec 2022 10:29:26 +0800 Subject: [PATCH] Added EmbeddingLookup 4d support internal Added EmbeddingLookup internal support for 4d lut input Type: New Feature Signed-off-by: Feiyue Chen --- src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c b/src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c index b3ae7cf..bcdf270 100644 --- a/src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c +++ b/src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c @@ -53,6 +53,13 @@ static void _reshape_tensor attr.size[2] = input->attr.size[1]; attr.dim_num = 3; } + else if (input->attr.dim_num == 4) + { + attr.size[0] = input->attr.size[0]; + attr.size[1] = input->attr.size[1] * input->attr.size[2]; + attr.size[2] = input->attr.size[3]; + attr.dim_num = 3; + } *output = vsi_nn_safe_reshape_tensor( input->t, (void*)attr.size, (vsi_size_t)attr.dim_num , sizeof(attr.size[0])); }