Home
last modified time | relevance | path

Searched refs:BatchNormalizationQueueDescriptor (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClBatchNormalizationFloatWorkload.hpp25 class ClBatchNormalizationFloatWorkload : public FloatWorkload<BatchNormalizationQueueDescriptor>
28 ClBatchNormalizationFloatWorkload(const BatchNormalizationQueueDescriptor& descriptor,
32 using FloatWorkload<BatchNormalizationQueueDescriptor>::FloatWorkload;
H A DClBatchNormalizationFloatWorkload.cpp55 const BatchNormalizationQueueDescriptor& descriptor, in ClBatchNormalizationFloatWorkload()
58 : FloatWorkload<BatchNormalizationQueueDescriptor>(descriptor, info) in ClBatchNormalizationFloatWorkload()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DRefBatchNormalizationWorkload.hpp14 class RefBatchNormalizationWorkload : public RefBaseWorkload<BatchNormalizationQueueDescriptor>
17 explicit RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& descriptor,
H A DBatchNormImpl.hpp16 void BatchNormImpl(const BatchNormalizationQueueDescriptor& data,
H A DBatchNormImpl.cpp18 void BatchNormImpl(const BatchNormalizationQueueDescriptor& data, in BatchNormImpl()
H A DRefBatchNormalizationWorkload.cpp16 …hNormalizationWorkload::RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& des… in RefBatchNormalizationWorkload()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonBatchNormalizationWorkload.hpp27 class NeonBatchNormalizationWorkload : public NeonBaseWorkload<BatchNormalizationQueueDescriptor>
30 NeonBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& descriptor,
H A DNeonBatchNormalizationWorkload.cpp60 const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) in NeonBatchNormalizationWorkload()
61 : NeonBaseWorkload<BatchNormalizationQueueDescriptor>(descriptor, info) in NeonBatchNormalizationWorkload()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DBatchNormalizationTestImpl.cpp80 armnn::BatchNormalizationQueueDescriptor descriptor; in BatchNormTestImpl()
162 armnn::BatchNormalizationQueueDescriptor data; in BatchNormTestNhwcImpl()
652 armnn::BatchNormalizationQueueDescriptor data; in CompareBatchNormTest()
672 armnn::BatchNormalizationQueueDescriptor refData = data; in CompareBatchNormTest()
/aosp_15_r20/external/armnn/include/armnn/backends/
H A DWorkloadData.hpp306 struct BatchNormalizationQueueDescriptor : QueueDescriptorWithParameters<BatchNormalizationDescript… struct
308 BatchNormalizationQueueDescriptor() in BatchNormalizationQueueDescriptor() argument
H A DWorkloadFactory.hpp94 …l std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& des…
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DBatchNormalizationLayer.cpp29 BatchNormalizationQueueDescriptor descriptor; in CreateWorkload()
/aosp_15_r20/external/armnn/src/backends/cl/test/
H A DOpenClTimerTest.cpp79 BatchNormalizationQueueDescriptor data;
H A DClCreateWorkloadTests.cpp185 BatchNormalizationQueueDescriptor queueDescriptor = workload->GetData(); in ClCreateBatchNormalizationWorkloadTest()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadFactoryBase.hpp54 …std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& /*des… in CreateBatchNormalization() argument
H A DWorkloadFactory.cpp1614 …auto batchNormQueueDescriptor = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&des… in CreateWorkload()
1963 const BatchNormalizationQueueDescriptor& /*descriptor*/, const WorkloadInfo& /*info*/) const in CreateBatchNormalization() argument
/aosp_15_r20/external/armnn/src/backends/neon/
H A DNeonWorkloadFactory.hpp76 …std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& descr…
H A DNeonWorkloadFactory.cpp163 = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor); in CreateWorkload()
658 const BatchNormalizationQueueDescriptor& descriptor, const WorkloadInfo& info) const in CreateBatchNormalization()
/aosp_15_r20/external/armnn/src/backends/reference/
H A DRefWorkloadFactory.cpp180 …auto batchNormQueueDescriptor = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&des… in CreateWorkload()
679 const BatchNormalizationQueueDescriptor& descriptor, in CreateBatchNormalization()
H A DRefWorkloadFactory.hpp90 …std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& descr…
/aosp_15_r20/external/armnn/src/backends/cl/
H A DClWorkloadFactory.hpp79 …std::unique_ptr<IWorkload> CreateBatchNormalization(const BatchNormalizationQueueDescriptor& descr…
H A DClWorkloadFactory.cpp276 = PolymorphicDowncast<const BatchNormalizationQueueDescriptor*>(&descriptor); in CreateWorkload()
804 const BatchNormalizationQueueDescriptor& descriptor, in CreateBatchNormalization()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DWorkloadDataValidation.cpp30 BatchNormalizationQueueDescriptor invalidData;
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonCreateWorkloadTests.cpp208 BatchNormalizationQueueDescriptor queueDescriptor = workload->GetData(); in NeonCreateBatchNormalizationWorkloadTest()
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp412 BatchNormalizationQueueDescriptor queueDescriptor = workload->GetData(); in CreateBatchNormalizationWorkloadTest()
485 BatchNormalizationQueueDescriptor queueDescriptor = workload->GetData(); in CreateBatchNormalizationWithBlobWorkloadTest()

12