Home
last modified time | relevance | path

Searched refs:GatherQueueDescriptor (Results 1 – 18 of 18) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DRefGatherWorkload.hpp19 class RefGatherWorkload : public RefBaseWorkload<GatherQueueDescriptor>
22 using RefBaseWorkload<GatherQueueDescriptor>::RefBaseWorkload;
/external/armnn/src/backends/neon/workloads/
DNeonGatherWorkload.hpp19 class NeonGatherWorkload : public NeonBaseWorkload<GatherQueueDescriptor>
22 NeonGatherWorkload(const GatherQueueDescriptor& descriptor, const WorkloadInfo& info);
DNeonGatherWorkload.cpp27 NeonGatherWorkload::NeonGatherWorkload(const GatherQueueDescriptor& descriptor, in NeonGatherWorkload()
29 : NeonBaseWorkload<GatherQueueDescriptor>(descriptor, info) in NeonGatherWorkload()
/external/armnn/src/backends/cl/workloads/
DClGatherWorkload.hpp19 class ClGatherWorkload : public ClBaseWorkload<GatherQueueDescriptor>
22 ClGatherWorkload(const GatherQueueDescriptor& descriptor,
DClGatherWorkload.cpp29 ClGatherWorkload::ClGatherWorkload(const GatherQueueDescriptor& descriptor, in ClGatherWorkload()
32 : ClBaseWorkload<GatherQueueDescriptor>(descriptor, info) in ClGatherWorkload()
/external/armnn/src/armnn/layers/
DGatherLayer.cpp23 GatherQueueDescriptor descriptor; in CreateWorkload()
/external/armnn/src/backends/backendsCommon/
DWorkloadFactoryBase.hpp135 std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& /*descriptor*/, in CreateGather() argument
DWorkloadFactory.cpp1739 … auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor); in CreateWorkload()
2094 std::unique_ptr<IWorkload> IWorkloadFactory::CreateGather(const GatherQueueDescriptor& /*descriptor… in CreateGather() argument
DWorkloadData.cpp2697 void GatherQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const in Validate()
/external/armnn/include/armnn/backends/
DWorkloadFactory.hpp204 virtual std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
DWorkloadData.hpp502 struct GatherQueueDescriptor : QueueDescriptorWithParameters<GatherDescriptor> struct
/external/armnn/src/backends/neon/
DNeonWorkloadFactory.hpp181 std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
DNeonWorkloadFactory.cpp394 … auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor); in CreateWorkload()
845 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateGather(const armnn::GatherQueueDescriptor& de… in CreateGather()
/external/armnn/src/backends/reference/
DRefWorkloadFactory.hpp200 std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
DRefWorkloadFactory.cpp352 … auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor); in CreateWorkload()
863 std::unique_ptr<IWorkload> RefWorkloadFactory::CreateGather(const GatherQueueDescriptor& descriptor, in CreateGather()
/external/armnn/src/backends/cl/
DClWorkloadFactory.hpp184 std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
DClWorkloadFactory.cpp539 … auto gatherQueueDescriptor = PolymorphicDowncast<const GatherQueueDescriptor*>(&descriptor); in CreateWorkload()
999 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateGather(const GatherQueueDescriptor& descriptor, in CreateGather()
/external/armnn/src/backends/backendsCommon/test/layerTests/
DGatherTestImpl.cpp43 armnn::GatherQueueDescriptor data; in GatherTestImpl()