24 #ifndef SRC_NEGEMMASSEMBLYDISPATCH_H 25 #define SRC_NEGEMMASSEMBLYDISPATCH_H 74 virtual void run() = 0;
120 std::unique_ptr<IFallback> _arm_gemm;
virtual bool is_configured() const =0
Base class for all functions.
bool reinterpret_input_as_3d
NEGEMMAssemblyDispatch(std::shared_ptr< IMemoryManager > memory_manager=nullptr, IWeightsManager *weights_manager=nullptr)
Constructor.
ActivationLayerInfo activation_info
~NEGEMMAssemblyDispatch()=default
Store the tensor's metadata.
void run() override
Run the kernels contained in the function.
Activation Layer Information class.
Interface for CPU tensor.
Copyright (c) 2017-2021 Arm Limited.
static bool is_activation_supported(const ActivationLayerInfo &activation)
Checks if activation is supported by the gemm assembly dispatcher.
void prepare() override
Prepare the function for executing.
NEGEMMAssemblyDispatch & operator=(const NEGEMMAssemblyDispatch &)=delete
Prevent instances of this class from being copied.
bool is_configured() const
Was the function successfully configured ?
GEMMLowp output stage info.
Padding and stride information class.
static Status validate(const ITensorInfo *a, const ITensorInfo *b, const ITensorInfo *c, const ITensorInfo *d, const AsmGemmInfo &info)
Indicates whether or not this function can be used to process the given parameters.
Weights manager interface to handle weights transformations.
GEMMLowpOutputStageInfo output_stage
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
void configure(const ITensor *a, const ITensor *b, const ITensor *c, ITensor *d, const AsmGemmInfo &info)
If supported create a Compute Library function else fallback to the arm_gemm function.
virtual ~IFallback()=default