diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-02 01:37:17 +0000 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-11-12 15:59:25 +0000 |
| commit | c0b6f76561580414f08633a804fc548ccad65659 (patch) | |
| tree | 4d46b7f479de04f799e29095392948aeb370c029 /src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp | |
| parent | 824061d9910ebb42cbe46b677c0b843db212c9a2 (diff) | |
| download | ComputeLibrary-c0b6f76561580414f08633a804fc548ccad65659.tar.gz | |
COMPMID-3776: Indirect GEMM
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: I51a1b0f098bc3a8c408c50c92221e4df3061e12c
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4343
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp')
| -rw-r--r-- | src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp index 134007b74c..854b6751c1 100644 --- a/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp +++ b/src/core/NEON/kernels/arm_gemm/kernels/a64_gemm_u8_4x4.hpp @@ -32,7 +32,7 @@ namespace arm_gemm { // Kernel definition void a64_gemm_u8_4x4(const uint8_t *Apanel, const uint8_t *Bpanel, uint32_t *Cpanel, int ablocks, int bblocks, int K); -class gemm_u8_4x4 { +class cls_a64_gemm_u8_4x4 { public: typedef uint8_t operand_type; typedef uint32_t result_type; @@ -64,10 +64,11 @@ public: // Use the standard fixed size transforms. StdTransformsFixed<operand_type, result_type, 4, 4, 16> transforms = {}; + StdTransformsFixed<operand_type, result_type, 4, 4, 16, true> transforms_quantized = {}; kern_type kernel = a64_gemm_u8_4x4; - gemm_u8_4x4(const CPUInfo *) { } + cls_a64_gemm_u8_4x4(const CPUInfo *) { } }; } // namespace arm_gemm |