|
25.11
|
#include <RefBatchMatMulWorkload.hpp>
Public Member Functions | |
| RefBatchMatMulWorkload (const BatchMatMulQueueDescriptor &descriptor, const WorkloadInfo &info) | |
| void | Execute () const override |
| Public Member Functions inherited from RefBaseWorkload< BatchMatMulQueueDescriptor > | |
| RefBaseWorkload (const BatchMatMulQueueDescriptor &descriptor, const WorkloadInfo &info) | |
| virtual bool | SupportsTensorHandleReplacement () const override |
| void | ReplaceInputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override |
| void | ReplaceOutputTensorHandle (ITensorHandle *tensorHandle, unsigned int slot) override |
| Public Member Functions inherited from BaseWorkload< BatchMatMulQueueDescriptor > | |
| BaseWorkload (const BatchMatMulQueueDescriptor &descriptor, const WorkloadInfo &info) | |
| virtual const std::string & | GetName () const override |
| void | PostAllocationConfigure () override |
| const BatchMatMulQueueDescriptor & | GetData () const |
| arm::pipe::ProfilingGuid | GetGuid () const final |
| Public Member Functions inherited from IWorkload | |
| virtual | ~IWorkload () |
| virtual void | RegisterDebugCallback (const DebugCallbackFunction &) |
| virtual armnn::Optional< armnn::MemoryRequirements > | GetMemoryRequirements () |
Additional Inherited Members | |
| Protected Attributes inherited from BaseWorkload< BatchMatMulQueueDescriptor > | |
| BatchMatMulQueueDescriptor | m_Data |
| const arm::pipe::ProfilingGuid | m_Guid |
| const std::string | m_Name |
Definition at line 16 of file RefBatchMatMulWorkload.hpp.
|
explicit |
Definition at line 15 of file RefBatchMatMulWorkload.cpp.
References armnn::info, and RefBaseWorkload< BatchMatMulQueueDescriptor >::RefBaseWorkload().
|
overridevirtual |
Implements IWorkload.
Definition at line 19 of file RefBatchMatMulWorkload.cpp.
References Execute(), and BaseWorkload< BatchMatMulQueueDescriptor >::m_Data.
Referenced by Execute().