Home
last modified time | relevance | path

Searched refs:BatchMatMulQueueDescriptor (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefBatchMatMulWorkload.hpp16 class RefBatchMatMulWorkload : public RefBaseWorkload<BatchMatMulQueueDescriptor>
19 explicit RefBatchMatMulWorkload(const BatchMatMulQueueDescriptor& descriptor,
H A DRefBatchMatMulWorkload.cpp15 RefBatchMatMulWorkload::RefBatchMatMulWorkload(const BatchMatMulQueueDescriptor& descriptor, const … in RefBatchMatMulWorkload()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonBatchMatMulWorkload.hpp22 class NeonBatchMatMulWorkload : public NeonBaseWorkload<BatchMatMulQueueDescriptor>
25 NeonBatchMatMulWorkload(const BatchMatMulQueueDescriptor& descriptor,
H A DNeonBatchMatMulWorkload.cpp103 const BatchMatMulQueueDescriptor& descriptor, const WorkloadInfo& info) in NeonBatchMatMulWorkload()
104 : NeonBaseWorkload<BatchMatMulQueueDescriptor>(descriptor, info) in NeonBatchMatMulWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClBatchMatMulWorkload.hpp21 class ClBatchMatMulWorkload : public ClBaseWorkload<BatchMatMulQueueDescriptor>
24 ClBatchMatMulWorkload(const BatchMatMulQueueDescriptor& descriptor,
H A DClBatchMatMulWorkload.cpp115 ClBatchMatMulWorkload::ClBatchMatMulWorkload(const BatchMatMulQueueDescriptor& descriptor, in ClBatchMatMulWorkload()
118 : ClBaseWorkload<BatchMatMulQueueDescriptor>(descriptor, info) in ClBatchMatMulWorkload()
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DBatchMatMulLayer.cpp20 BatchMatMulQueueDescriptor descriptor; in CreateWorkload()
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DWorkloadData.hpp748 struct BatchMatMulQueueDescriptor : QueueDescriptorWithParameters<BatchMatMulDescriptor> struct
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefWorkloadFactory.cpp175 …auto batchMatMulQueueDescriptor = PolymorphicDowncast<const BatchMatMulQueueDescriptor*>(&descript… in CreateWorkload()
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonWorkloadFactory.cpp157 …auto batchMatMulQueueDescriptor = PolymorphicDowncast<const BatchMatMulQueueDescriptor*>(&descript… in CreateWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClWorkloadFactory.cpp270 …auto batchMatMulQueueDescriptor = PolymorphicDowncast<const BatchMatMulQueueDescriptor*>(&descript… in CreateWorkload()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DBatchMatMulTestImpl.cpp38 armnn::BatchMatMulQueueDescriptor queueDescriptor; in BatchMatMulTestImpl()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp4079 void BatchMatMulQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const in Validate()