diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions')
4 files changed, 4 insertions, 0 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMM.h b/arm_compute/runtime/NEON/functions/NEGEMM.h index 5279995be4..e2263c2307 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMM.h +++ b/arm_compute/runtime/NEON/functions/NEGEMM.h @@ -83,6 +83,7 @@ private: Tensor _tmp_a; Tensor _tmp_b; Tensor _workspace; + Tensor _B_pretransposed; bool _run_vector_matrix_multiplication; bool _run_addition; bool _is_first_run; diff --git a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h index 24e23f133a..752693188c 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h @@ -185,6 +185,7 @@ private: Tensor _gemm_output; Tensor _tmp_output; Tensor _workspace; + Tensor _B_pretransposed; bool _append_bias; bool _is_fully_connected_convolution; diff --git a/arm_compute/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.h b/arm_compute/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.h index f09c94e726..11ca1bc313 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMLowpAssemblyMatrixMultiplyCore.h @@ -67,6 +67,7 @@ private: Tensor _tmp_a; Tensor _tmp_b; Tensor _workspace; + Tensor _B_pretransposed; }; } #endif /*__ARM_COMPUTE_NEGEMMLOWPASSEMBLYMATRIXMULTIPLYCORE_H__ */ diff --git a/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h b/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h index 3c9fb0ea5f..adcddb8263 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h @@ -102,6 +102,7 @@ private: Tensor _tmp_a; Tensor _tmp_b; Tensor _workspace; + Tensor _B_pretranspose; int32_t _a_offset; int32_t _b_offset; bool _run_vector_matrix_multiplication; |