Compute Library
 22.05
arm_gemm.hpp File Reference
#include <cstring>
#include <memory>
#include <vector>
#include "arm_gemm_local.hpp"
#include "gemm_common.hpp"

Go to the source code of this file.

Data Structures

struct  KernelDescription
 
struct  GemmConfig
 
struct  Activation
 
struct  GemmArgs
 
struct  Requantize32
 
struct  Nothing
 

Namespaces

 arm_gemm
 

Typedefs

template<typename Top , typename Tret >
using UniqueGemmCommon = std::unique_ptr< GemmCommon< Top, Tret > >
 

Enumerations

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
}
 

Functions

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)