colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added EmbeddingLookup 4d support internal 

Added EmbeddingLookup internal support for 4d lut input

Type: New Feature
Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Feiyue Chen 2022-12-16 10:29:26 +08:00 committed by Sven
parent 06b88e7940
commit ac4517b5c1
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/tim/vx/internal/src/ops/vsi_nn_op_embedding_lookup.c
View File

@ -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]));
}