aboutsummaryrefslogtreecommitdiff
path: root/reference_model/src/ops/tensor_ops.cc
diff options
context:
space:
mode:
authorWon Jeon <won.jeon@arm.com>2024-03-13 19:18:08 +0000
committerWon Jeon <won.jeon@arm.com>2024-03-15 21:24:38 +0000
commite67115ef82bcba0718dcbd75cc8411985001b7cc (patch)
treece6896be8d6e25517d913517ef4f5966df787f16 /reference_model/src/ops/tensor_ops.cc
parent00dc22eea26159661074216688a2c0b9cfb9d709 (diff)
downloadreference_model-e67115ef82bcba0718dcbd75cc8411985001b7cc.tar.gz
Add missing FP8 definition and instantiation for FULLY_CONNNECTED op
Signed-off-by: Won Jeon <won.jeon@arm.com> Change-Id: I7efc03f16136b2b74f8dfef79a6e7e147c93f4ed
Diffstat (limited to 'reference_model/src/ops/tensor_ops.cc')
-rw-r--r--reference_model/src/ops/tensor_ops.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc
index 609265c..7bd249b 100644
--- a/reference_model/src/ops/tensor_ops.cc
+++ b/reference_model/src/ops/tensor_ops.cc
@@ -2213,6 +2213,8 @@ DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT8, INT4, INT32);
DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT8, INT8, INT32);
DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, INT16, INT8, INT48);
DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP64, FP64, FP64);
+DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP8E4M3, FP8E4M3, FP16);
+DEF_INSTANTIATE_THREE_TYPE(OpFullyConnected, FP8E5M2, FP8E5M2, FP16);
DEF_INSTANTIATE_TWO_TYPE(OpMatMul, INT8, INT32);
DEF_INSTANTIATE_TWO_TYPE(OpMatMul, INT16, INT48);