commit 44179c372eea9f17c96cbf50ee383e57e14d70a6
author Narumol Prangnawarat <narumol.prangnawarat@arm.com> Wed Mar 11 14:51:27 2020 +0000
committer Narumol Prangnawarat <narumol.prangnawarat@arm.com> Fri Mar 13 09:49:42 2020 +0000
tree 2a2971c2db67426107b21d9a045cfa46a4a1663a
parent e9b5d2989abc8008df7ff3ea287ee896ee1121a6

IVGCVSW-4511 Add BFloat16 to RefLayerSupport and unit tests

Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: Ifaae4d5aac468ba927b2c6a4bf31b8c8522aeb2e

src/backends/reference/workloads/RefTransposeWorkload.cpp

diff --git a/src/backends/reference/workloads/RefTransposeWorkload.cpp b/src/backends/reference/workloads/RefTransposeWorkload.cpp
index 6bdfb21..242668b 100644
--- a/src/backends/reference/workloads/RefTransposeWorkload.cpp
+++ b/src/backends/reference/workloads/RefTransposeWorkload.cpp
@@ -27,6 +27,7 @@
     armnnUtils::Transpose(GetTensorInfo(src).GetShape(), mappings, src->Map(), dst->Map(), sizeof(T));
 }
 
+template class RefTransposeWorkload<DataType::BFloat16>;
 template class RefTransposeWorkload<DataType::Float16>;
 template class RefTransposeWorkload<DataType::Float32>;
 template class RefTransposeWorkload<DataType::QAsymmU8>;