From 6b47809e7d6c55d20a05d863ce2f09159f381f85 Mon Sep 17 00:00:00 2001 From: Samuel Yap Date: Wed, 6 Jul 2022 15:36:03 +0100 Subject: IVGCVSW-7109: Add Batch MatMul front end support - Reference * Descriptors added for BatchMatMul * Layer definition added * Input validation added (will likely change when opt. param support comes in) * Ref workload implementation for BatchMatMul added (will also change with opt. param support) * Ref layer tests made for BatchMatMul * CMake and other build files updated Signed-off-by: Samuel Yap Change-Id: Ic885301da543ee0fbe7922b85e7f9658c4efc617 --- docs/02_operator_list.dox | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) (limited to 'docs/02_operator_list.dox') diff --git a/docs/02_operator_list.dox b/docs/02_operator_list.dox index 960428999e..658aa07d1d 100644 --- a/docs/02_operator_list.dox +++ b/docs/02_operator_list.dox @@ -267,6 +267,48 @@ where N = batches, C = channels, H = height, W = width FLOAT16 FLOAT32 + + BatchMatMulLayer + Layer to perform batch matrix multiplication. + + + CpuRef + + + + +
+
BFLOAT16 +
FLOAT16 +
FLOAT32 +
QASYMMS8 +
QASYMMU8 +
QSYMMS16 +
+ + CpuAcc + + + + + + GpuAcc + + + + BatchNormalizationLayer Layer to perform batch normalization. -- cgit v1.2.1