diff options
Diffstat (limited to 'src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp')
| -rw-r--r-- | src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp | 15 |
1 files changed, 11 insertions, 4 deletions
diff --git a/src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp b/src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp index 9c1f6f3c0f..e12a312280 100644 --- a/src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp +++ b/src/cpu/kernels/gemm_matrix_mul/generic/neon/fp32.cpp @@ -28,9 +28,16 @@ namespace arm_compute { namespace cpu { -void neon_fp32_gemm_matrix_mul(const ITensor *lhs, const ITensor *rhs, ITensor *dst, const Window &window, const ThreadInfo &info, float alpha, const bool is_dst_vector) +void neon_fp32_gemm_matrix_mul(const ITensor *lhs, + const ITensor *rhs, + ITensor *dst, + const Window &window, + const ThreadInfo &info, + float alpha, + const bool is_dst_vector) { - return (is_dst_vector) ? vector_matrix_multiply_f32(lhs, rhs, dst, window, info, alpha) : matrix_matrix_multiply_f32(lhs, rhs, dst, window, info, alpha); + return (is_dst_vector) ? vector_matrix_multiply_f32(lhs, rhs, dst, window, info, alpha) + : matrix_matrix_multiply_f32(lhs, rhs, dst, window, info, alpha); } -} // namespce cpu -} // namespace arm_compute
\ No newline at end of file +} // namespace cpu +} // namespace arm_compute |