VeriSilicon / tflite-vx-delegate

Tensorflow Lite external delegate based on TIM-VX
MIT License
41 stars 23 forks source link

Fixed bug when fc output is not as required shape #164

Closed chenfeiyue-cfy closed 1 year ago

chenfeiyue-cfy commented 1 year ago

Backgraoud: mobilebert_quant model crashed when fc link to a 3-dimensional transpose

Solution: added reshape op after fc if the required output shape is over 2-dimension

Type: Bug Fix Issue: bugzilla id 33962 Signed-off-by: Feiyue Chen Feiyue.Chen@verisilicon.com