/** * \file dnn/src/arm_common/matrix_mul/int8/gemv.h * MegEngine is Licensed under the Apache License, Version 2.0 (the "License") * * Copyright (c) 2014-2020 Megvii Inc. All rights reserved. * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. */ #pragma once #include #include #if !__ARM_FEATURE_DOTPROD namespace megdnn { namespace arm_common { namespace matmul { bool is_gemv_like_preferred_int8(bool transposeA, bool transposeB, size_t M, size_t N, size_t K, size_t LDA, size_t LDB, size_t LDC); void gemv_like_int8(const int8_t* __restrict A, const int8_t* __restrict B, int32_t* __restrict C, size_t M, size_t N, size_t K, size_t Astride, size_t Bstride, size_t Cstride); } // namespace matmul } // namespace arm_common } // namespace megdnn #endif // vim: syntax=cpp.doxygen