24.04
|
Go to the documentation of this file.
37 const ITensor *lhs{
nullptr};
38 const ITensor *rhs{
nullptr};
39 ITensor *output{
nullptr};
40 std::unique_ptr<cpu::CpuMatMul> op{
nullptr};
41 MemoryGroup memory_group{};
42 WorkspaceData<Tensor> workspace_tensors{};
43 ITensorPack run_pack{};
61 _impl->output = output;
64 _impl->op = std::make_unique<cpu::CpuMatMul>();
67 _impl->workspace_tensors = manage_workspace<Tensor>(_impl->op->workspace(), _impl->memory_group, _impl->run_pack);
83 _impl->op->run(_impl->run_pack);
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(ITensor *lhs, ITensor *rhs, ITensor *dst, const MatMulInfo &info, const CpuMatMulSettings &settings, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialize.
Interface for CPU tensor.
Activation Layer Information class.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
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 of NEMatMul.
Memory group resources scope handling class.
Settings for MatMul Cpu implementation.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
void run() override
Run the kernels contained in the function.
Class for holding information related to matrix multiplication function.