diff options
Diffstat (limited to 'src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h')
-rw-r--r-- | src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h index 210f22b109..c2ae2a67f4 100644 --- a/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h +++ b/src/gpu/cl/kernels/helpers/MatMulKernelHelpers.h @@ -44,7 +44,8 @@ namespace kernels * * @return true if the shapes and matmul kernel info matches */ -Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorShape &rhs_shape, +Status validate_matmul_input_shapes(const TensorShape &lhs_shape, + const TensorShape &rhs_shape, const MatMulKernelInfo &matmul_kernel_info); /** Validate and configure window for Matmul MMUL kernels @@ -58,9 +59,12 @@ Status validate_matmul_input_shapes(const TensorShape &lhs_shape, const TensorSh * * @return a pair of Status and Window object */ -std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, - const ITensorInfo *rhs, const ITensorInfo *dst, const MatMulKernelInfo &matmul_kernel_info, - int mmul_m0, int mmul_n0); +std::pair<Status, Window> validate_and_configure_window_for_mmul_kernels(const ITensorInfo *lhs, + const ITensorInfo *rhs, + const ITensorInfo *dst, + const MatMulKernelInfo &matmul_kernel_info, + int mmul_m0, + int mmul_n0); } // namespace kernels } // namespace opencl } // namespace arm_compute |