Searched refs:SoftmaxQueueDescriptor (Results 1 – 22 of 22) sorted by relevance
| /external/armnn/src/backends/reference/workloads/ |
| D | RefSoftmaxWorkload.hpp | 14 class RefSoftmaxWorkload : public RefBaseWorkload<SoftmaxQueueDescriptor>
17 using RefBaseWorkload<SoftmaxQueueDescriptor>::RefBaseWorkload;
|
| /external/armnn/src/backends/neon/workloads/ |
| D | NeonSoftmaxWorkload.hpp | 24 class NeonSoftmaxWorkload : public NeonBaseWorkload<SoftmaxQueueDescriptor>
27 NeonSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor, const WorkloadInfo& info,
|
| D | NeonSoftmaxWorkload.cpp | 33 NeonSoftmaxWorkload::NeonSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor, in NeonSoftmaxWorkload()
35 : NeonBaseWorkload<SoftmaxQueueDescriptor>(descriptor, info) in NeonSoftmaxWorkload()
|
| /external/armnn/src/backends/cl/workloads/ |
| D | ClSoftmaxWorkload.hpp | 23 class ClSoftmaxWorkload : public ClBaseWorkload<SoftmaxQueueDescriptor>
26 ClSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor,
|
| D | ClSoftmaxWorkload.cpp | 28 ClSoftmaxWorkload::ClSoftmaxWorkload(const SoftmaxQueueDescriptor& descriptor, in ClSoftmaxWorkload()
32 : ClBaseWorkload<SoftmaxQueueDescriptor>(descriptor, info) in ClSoftmaxWorkload()
|
| /external/armnn/src/armnn/layers/ |
| D | SoftmaxLayer.cpp | 23 SoftmaxQueueDescriptor descriptor; in CreateWorkload()
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | SoftmaxTestImpl.cpp | 93 armnn::SoftmaxQueueDescriptor data; in SimpleSoftmaxBaseTestImpl()
269 armnn::SoftmaxQueueDescriptor data; in CompareSoftmaxTestImpl()
281 armnn::SoftmaxQueueDescriptor refData = data; in CompareSoftmaxTestImpl()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadFactoryBase.hpp | 239 std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& /*descriptor*/, in CreateSoftmax() argument
|
| D | WorkloadFactory.cpp | 1887 … auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor); in CreateWorkload()
2280 std::unique_ptr<IWorkload> IWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& /*descript… in CreateSoftmax() argument
|
| D | WorkloadData.cpp | 747 void SoftmaxQueueDescriptor::Validate(const WorkloadInfo& workloadInfo) const in Validate()
|
| /external/armnn/include/armnn/backends/ |
| D | WorkloadFactory.hpp | 360 virtual std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
|
| D | WorkloadData.hpp | 105 struct SoftmaxQueueDescriptor : QueueDescriptorWithParameters<SoftmaxDescriptor> struct
|
| /external/armnn/src/backends/neon/ |
| D | NeonWorkloadFactory.hpp | 321 std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
|
| D | NeonWorkloadFactory.cpp | 567 … auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor); in CreateWorkload()
1034 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descrip… in CreateSoftmax()
|
| /external/armnn/src/backends/reference/ |
| D | RefWorkloadFactory.hpp | 345 std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
|
| D | RefWorkloadFactory.cpp | 575 … auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor); in CreateWorkload()
1107 std::unique_ptr<IWorkload> RefWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descript… in CreateSoftmax()
|
| /external/armnn/src/backends/cl/ |
| D | ClWorkloadFactory.hpp | 324 std::unique_ptr<IWorkload> CreateSoftmax(const SoftmaxQueueDescriptor& descriptor,
|
| D | ClWorkloadFactory.cpp | 722 … auto softmaxQueueDescriptor = PolymorphicDowncast<const SoftmaxQueueDescriptor*>(&descriptor); in CreateWorkload()
1189 std::unique_ptr<IWorkload> ClWorkloadFactory::CreateSoftmax(const SoftmaxQueueDescriptor& descripto… in CreateSoftmax()
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 120 SoftmaxQueueDescriptor invalidData;
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonCreateWorkloadTests.cpp | 616 SoftmaxQueueDescriptor queueDescriptor = workload->GetData(); in NeonCreateSoftmaxWorkloadTest()
|
| /external/armnn/src/backends/cl/test/ |
| D | ClCreateWorkloadTests.cpp | 755 SoftmaxQueueDescriptor queueDescriptor = workload->GetData(); in ClSoftmaxWorkloadTest()
|
| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 1558 SoftmaxQueueDescriptor queueDescriptor = workload->GetData(); in CreateSoftmaxWorkloadTest()
|