From 6bf1613c5894d81849dd12b5be6145c1f24caca2 Mon Sep 17 00:00:00 2001 From: James Ward Date: Wed, 8 Sep 2021 11:14:20 +0100 Subject: MLBEDSW-5162 MLCE: Vela [3.1.0] falling to run with yolov4_int8.tflite * fix indices for tflite mapping of EXP operator * fix indices for tflite mapping of Transpose operator * ensure read offset after slice is aligned to 16 bytes for NHCWB16 or force linear format * add unit test to ensure mapping of indices is consistent across TFLite, TOSA and NNG Signed-off-by: James Ward Change-Id: I17b6e44bc06853325d5eea62a558418ee1ebefe8 --- ethosu/vela/operation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/operation.py') diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 1e733d56..1707d318 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -285,7 +285,7 @@ class Op(Enum): Tanh = OperatorInfo(indices=NNG_IFM_INDICES) Tile = OperatorInfo() TopKV2 = OperatorInfo() - Transpose = OperatorInfo(indices=NNG_IFM_INDICES) + Transpose = OperatorInfo(indices=NNG_IFM_IFM2_INDICES) UnidirectionalSequenceLstm = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) UnidirectionalSequenceRnn = OperatorInfo(block_type=NpuBlockType.VectorProduct, indices=NNG_IFM_WEIGHTS_INDICES) Unique = OperatorInfo() -- cgit v1.2.1