#include <cstring>
#include <memory>
#include <vector>
#include "arm_gemm_local.hpp"
#include "gemm_common.hpp"
Go to the source code of this file.
|
template<typename Top , typename Tret > |
using | UniqueGemmCommon = std::unique_ptr< GemmCommon< Top, Tret > > |
|
|
enum | GemmMethod {
DEFAULT,
GEMV_BATCHED,
GEMV_PRETRANSPOSED,
GEMV_NATIVE_TRANSPOSED,
GEMM_NATIVE,
GEMM_HYBRID,
GEMM_INTERLEAVED,
GEMM_INTERLEAVED_2D,
QUANTIZE_WRAPPER,
QUANTIZE_WRAPPER_2D,
GEMM_HYBRID_QUANTIZED
} |
|
|
template<typename Top , typename Tret , class OutputStage = Nothing> |
KernelDescription | get_gemm_method (const GemmArgs &args, const OutputStage &={}) |
|
template<typename Top , typename Tret , class OutputStage > |
UniqueGemmCommon< Top, Tret > | gemm (const GemmArgs &args, const OutputStage &os) |
|
template<typename Top , typename Tret , class OutputStage > |
std::vector< KernelDescription > | get_compatible_kernels (const GemmArgs &args, const OutputStage &os) |
|
template<typename Top , typename Tret , class OutputStage > |
bool | has_opt_gemm (const GemmArgs &args, const OutputStage &os) |
|