VeriSilicon / tflite-vx-delegate

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

Add TF MultiHeadAttention test model #163

Closed sunshinemyson closed 1 year ago

sunshinemyson commented 1 year ago

Remove unreasonable check for batchMatMul Verified with tensorflow v2.11.0

Signed-off-by: xiang.zhang xiang.zhang@verisilicon.com