diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h | 8 |
1 files changed, 3 insertions, 5 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h b/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h index f32eb3c757..96ac7bb7e0 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h @@ -30,7 +30,7 @@ #include "arm_compute/runtime/IFunction.h" #include "arm_compute/runtime/IMemoryManager.h" #include "arm_compute/runtime/MemoryGroup.h" -#include "arm_compute/runtime/NEON/AssemblyHelper.h" +#include "arm_compute/runtime/NEON/functions/NEGEMMAssemblyDispatch.h" #include "arm_compute/runtime/Tensor.h" #include <memory> @@ -98,8 +98,8 @@ public: private: MemoryGroup _memory_group; - AssemblyKernelGlueU8U32 _asm_glue_unsigned; - AssemblyKernelGlueS8S32 _asm_glue_signed; + NEGEMMAssemblyDispatchU8U32 _asm_glue_unsigned; + NEGEMMAssemblyDispatchS8S32 _asm_glue_signed; std::unique_ptr<INEKernel> _mm_kernel; std::unique_ptr<INEKernel> _mtx_a_reshape_kernel; std::unique_ptr<INEKernel> _mtx_b_reshape_kernel; @@ -110,8 +110,6 @@ private: Tensor _vector_sum_row; Tensor _tmp_a; Tensor _tmp_b; - Tensor _workspace; - Tensor _B_pretranspose; const ITensor *_original_b; int32_t _a_offset; int32_t _b_offset; |