24.04
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CPU_OPERATORS_CPUMATMUL_H
25 #define ACL_SRC_CPU_OPERATORS_CPUMATMUL_H
38 class CpuMatMulSettings;
94 enum InternalTensorIdx
103 std::unique_ptr<kernels::CpuTransposeKernel> _transpose_kernel_lhs{
nullptr};
104 std::unique_ptr<kernels::CpuTransposeKernel> _transpose_kernel_rhs{
nullptr};
105 std::unique_ptr<CpuGemmAssemblyDispatch> _asm_glue{
nullptr};
117 bool _adj_lhs{
false};
118 bool _adj_rhs{
false};
119 bool _fast_math{
false};
126 #endif // ACL_SRC_CPU_OPERATORS_CPUMATMUL_H
std::vector< MemoryInfo > MemoryRequirements
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
static Status validate(const ITensorInfo *lhs, const ITensorInfo *rhs, const ITensorInfo *dst, const MatMulInfo &info, const CpuMatMulSettings &settings, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
void configure(ITensorInfo *lhs, ITensorInfo *rhs, ITensorInfo *dst, const MatMulInfo &info, const CpuMatMulSettings &settings, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Configure operator for a given list of arguments.
Activation Layer Information class.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Basic interface for functions which have a single async CPU kernel.
Store the tensor's metadata.
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuMatMul)
Settings for MatMul Cpu implementation.
Copyright (c) 2017-2024 Arm Limited.
Function to execute MatMul Operation.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
Class for holding information related to matrix multiplication function.