| /external/armnn/src/armnnTestUtils/ |
| D | CreateWorkload.hpp | 92 ActivationQueueDescriptor queueDescriptor = workload->GetData(); in CreateActivationWorkloadTest() local
93 CHECK(queueDescriptor.m_Inputs.size() == 1); in CreateActivationWorkloadTest()
94 CHECK(queueDescriptor.m_Outputs.size() == 1); in CreateActivationWorkloadTest()
95 CHECK(queueDescriptor.m_Parameters.m_A == 3.5f); in CreateActivationWorkloadTest()
96 CHECK(queueDescriptor.m_Parameters.m_B == -10.0f); in CreateActivationWorkloadTest()
97 CHECK((queueDescriptor.m_Parameters.m_Function == ActivationFunction::ReLu)); in CreateActivationWorkloadTest()
128 auto queueDescriptor = workload->GetData(); in CreateElementwiseWorkloadTest() local
129 CHECK(queueDescriptor.m_Inputs.size() == 2); in CreateElementwiseWorkloadTest()
130 CHECK(queueDescriptor.m_Outputs.size() == 1); in CreateElementwiseWorkloadTest()
162 auto queueDescriptor = workload->GetData(); in CreateElementwiseBinaryWorkloadTest() local
[all …]
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonCreateWorkloadTests.cpp | 79 ActivationQueueDescriptor queueDescriptor = workload->GetData(); in NeonCreateActivationWorkloadTest() local
80 auto inputHandle = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Inputs[0]); in NeonCreateActivationWorkloadTest()
81 auto outputHandle = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Outputs[0]); in NeonCreateActivationWorkloadTest()
110 DescriptorType queueDescriptor = workload->GetData(); in NeonCreateElementwiseWorkloadTest() local
111 auto inputHandle1 = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Inputs[0]); in NeonCreateElementwiseWorkloadTest()
112 auto inputHandle2 = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Inputs[1]); in NeonCreateElementwiseWorkloadTest()
113 auto outputHandle = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Outputs[0]); in NeonCreateElementwiseWorkloadTest()
208 BatchNormalizationQueueDescriptor queueDescriptor = workload->GetData(); in NeonCreateBatchNormalizationWorkloadTest() local
209 auto inputHandle = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Inputs[0]); in NeonCreateBatchNormalizationWorkloadTest()
210 auto outputHandle = PolymorphicDowncast<IAclTensorHandle*>(queueDescriptor.m_Outputs[0]); in NeonCreateBatchNormalizationWorkloadTest()
[all …]
|
| /external/armnn/src/backends/cl/test/ |
| D | ClCreateWorkloadTests.cpp | 47 ActivationQueueDescriptor queueDescriptor = workload->GetData(); in ClCreateActivationWorkloadTest() local
48 auto inputHandle = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Inputs[0]); in ClCreateActivationWorkloadTest()
49 auto outputHandle = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Outputs[0]); in ClCreateActivationWorkloadTest()
79 auto queueDescriptor = workload->GetData(); in ClCreateElementwiseWorkloadTest() local
80 auto inputHandle1 = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Inputs[0]); in ClCreateElementwiseWorkloadTest()
81 auto inputHandle2 = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Inputs[1]); in ClCreateElementwiseWorkloadTest()
82 auto outputHandle = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Outputs[0]); in ClCreateElementwiseWorkloadTest()
156 DescriptorType queueDescriptor = workload->GetData(); in ClCreateElementwiseUnaryWorkloadTest() local
158 auto inputHandle = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Inputs[0]); in ClCreateElementwiseUnaryWorkloadTest()
159 auto outputHandle = PolymorphicDowncast<IClTensorHandle*>(queueDescriptor.m_Outputs[0]); in ClCreateElementwiseUnaryWorkloadTest()
[all …]
|
| /external/armnn/src/backends/reference/test/ |
| D | RefCreateWorkloadTests.cpp | 22 auto queueDescriptor = workload->GetData(); in CheckInputOutput() local
23 auto inputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]); in CheckInputOutput()
24 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]); in CheckInputOutput()
35 auto queueDescriptor = workload->GetData(); in CheckInputsOutput() local
36 auto inputHandle0 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]); in CheckInputsOutput()
37 auto inputHandle1 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[1]); in CheckInputsOutput()
38 auto outputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]); in CheckInputsOutput()
706 SplitterQueueDescriptor queueDescriptor = workload->GetData(); in RefCreateSplitterWorkloadTest() local
707 auto inputHandle = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Inputs[0]); in RefCreateSplitterWorkloadTest()
710 auto outputHandle0 = PolymorphicDowncast<RefTensorHandle*>(queueDescriptor.m_Outputs[0]); in RefCreateSplitterWorkloadTest()
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | WorkloadDataValidation.cpp | 668 Convolution2dQueueDescriptor queueDescriptor; variable
669 queueDescriptor.m_Parameters.m_BiasEnabled = true;
672 AddInputToWorkload(queueDescriptor, workloadInfo, inputInfo, nullptr);
673 AddInputToWorkload(queueDescriptor, workloadInfo, weightInfo, nullptr);
674 AddOutputToWorkload(queueDescriptor, workloadInfo, outputInfo, nullptr);
680 AddInputToWorkload(queueDescriptor, workloadInfo, biasInfo1, nullptr);
682 CHECK_NOTHROW(queueDescriptor.Validate(workloadInfo));
713 Convolution2dQueueDescriptor queueDescriptor; variable
714 queueDescriptor.m_Parameters.m_BiasEnabled = true;
717 AddInputToWorkload(queueDescriptor, workloadInfo, inputInfo, nullptr);
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | TransposeConvolution2dTestImpl.cpp | 73 TransposeConvolution2dQueueDescriptor queueDescriptor; in TransposeConvolution2dTestImpl() local
74 queueDescriptor.m_Parameters = descriptor; in TransposeConvolution2dTestImpl()
75 queueDescriptor.m_Weight = &weightsTensor; in TransposeConvolution2dTestImpl()
84 queueDescriptor.m_Bias = biasesTensor.get(); in TransposeConvolution2dTestImpl()
95 AddInputToWorkload(queueDescriptor, workloadInfo, input.first, inputHandle.get()); in TransposeConvolution2dTestImpl()
96 AddOutputToWorkload(queueDescriptor, workloadInfo, output.first, outputHandle.get()); in TransposeConvolution2dTestImpl()
99 …loadFactory.CreateWorkload(armnn::LayerType::TransposeConvolution2d, queueDescriptor, workloadInfo… in TransposeConvolution2dTestImpl()
653 TransposeConvolution2dQueueDescriptor queueDescriptor; in TransposeConvolution2dPerAxisQuantTest() local
654 queueDescriptor.m_Parameters = descriptor; in TransposeConvolution2dPerAxisQuantTest()
655 queueDescriptor.m_Weight = &weightTensor; in TransposeConvolution2dPerAxisQuantTest()
[all …]
|
| D | AdditionTestImpl.cpp | 578 armnn::Pooling2dQueueDescriptor queueDescriptor; in AdditionAfterMaxPoolTest() local
579 queueDescriptor.m_Parameters = descriptor; in AdditionAfterMaxPoolTest()
581 …AddInputToWorkload(queueDescriptor, workloadInfo, poolingInputTensorInfo, poolingInputHandle.get()… in AdditionAfterMaxPoolTest()
582 …AddOutputToWorkload(queueDescriptor, workloadInfo, poolingOutputTensorInfo, poolingOutputHandle.ge… in AdditionAfterMaxPoolTest()
586 queueDescriptor, in AdditionAfterMaxPoolTest()
|
| D | ConcatTestImpl.cpp | 144 PermuteQueueDescriptor queueDescriptor; in PermuteTensorData() local
145 queueDescriptor.m_Parameters = PermuteDescriptor{mappings}; in PermuteTensorData()
147 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in PermuteTensorData()
148 AddOutputToWorkload(queueDescriptor, workloadInfo, outputTensorInfo, outputHandle.get()); in PermuteTensorData()
151 queueDescriptor, in PermuteTensorData()
336 ConcatQueueDescriptor queueDescriptor; in Concatenate() local
338 queueDescriptor.m_Parameters = viewsDescriptor; in Concatenate()
342 queueDescriptor.m_ViewOrigins.reserve(viewsDescriptor.GetNumViews()); in Concatenate()
345 …queueDescriptor.m_ViewOrigins.emplace_back(std::vector<unsigned int>(viewsDescriptor.GetViewOrigin… in Concatenate()
360 … queueDescriptor.m_ViewOrigins[i].m_Origin.data()) : in Concatenate()
[all …]
|
| D | BatchMatMulTestImpl.cpp | 38 armnn::BatchMatMulQueueDescriptor queueDescriptor; in BatchMatMulTestImpl() local
39 queueDescriptor.m_Parameters = descriptor; in BatchMatMulTestImpl()
42 AddInputToWorkload(queueDescriptor, workloadInfo, inputXInfo, inputXHandle.get()); in BatchMatMulTestImpl()
43 AddInputToWorkload(queueDescriptor, workloadInfo, inputYInfo, inputYHandle.get()); in BatchMatMulTestImpl()
44 AddOutputToWorkload(queueDescriptor, workloadInfo, outputInfo, outputHandle.get()); in BatchMatMulTestImpl()
46 …auto workload = workloadFactory.CreateWorkload(armnn::LayerType::BatchMatMul, queueDescriptor, wor… in BatchMatMulTestImpl()
|
| D | Pooling3dTestImpl.cpp | 95 armnn::Pooling3dQueueDescriptor queueDescriptor; in SimplePooling3dTestImpl() local
96 queueDescriptor.m_Parameters = descriptor; in SimplePooling3dTestImpl()
97 queueDescriptor.m_Parameters.m_DataLayout = dataLayout; in SimplePooling3dTestImpl()
100 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in SimplePooling3dTestImpl()
101 AddOutputToWorkload(queueDescriptor, workloadInfo, outputTensorInfo, outputHandle.get()); in SimplePooling3dTestImpl()
109 queueDescriptor.m_Parameters, in SimplePooling3dTestImpl()
117 queueDescriptor, in SimplePooling3dTestImpl()
|
| D | Conv2dTestImpl.cpp | 3634 Convolution2dQueueDescriptor queueDescriptor; in Convolution2dPerAxisQuantTest() local
3635 queueDescriptor.m_Parameters = descriptor; in Convolution2dPerAxisQuantTest()
3637 AddInputToWorkload(queueDescriptor, workloadInfo, inputInfo, inputHandle.get()); in Convolution2dPerAxisQuantTest()
3638 AddInputToWorkload(queueDescriptor, workloadInfo, kernelInfo, weightsHandle.get()); in Convolution2dPerAxisQuantTest()
3643 AddInputToWorkload(queueDescriptor, workloadInfo, biasInfo, biasHandle.get()); in Convolution2dPerAxisQuantTest()
3646 AddOutputToWorkload(queueDescriptor, workloadInfo, outputInfo, outputHandle.get()); in Convolution2dPerAxisQuantTest()
3649 queueDescriptor, in Convolution2dPerAxisQuantTest()
3923 DepthwiseConvolution2dQueueDescriptor queueDescriptor; in DepthwiseConvolution2dPerAxisQuantTest() local
3926 AddInputToWorkload(queueDescriptor, workloadInfo, inputInfo, inputHandle.get()); in DepthwiseConvolution2dPerAxisQuantTest()
3927 AddInputToWorkload(queueDescriptor, workloadInfo, kernelInfo, weightsHandle.get()); in DepthwiseConvolution2dPerAxisQuantTest()
[all …]
|
| D | Pooling2dTestImpl.cpp | 82 armnn::Pooling2dQueueDescriptor queueDescriptor; in SimplePooling2dTestImpl() local
83 queueDescriptor.m_Parameters = descriptor; in SimplePooling2dTestImpl()
84 queueDescriptor.m_Parameters.m_DataLayout = dataLayout; in SimplePooling2dTestImpl()
87 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in SimplePooling2dTestImpl()
88 AddOutputToWorkload(queueDescriptor, workloadInfo, outputTensorInfo, outputHandle.get()); in SimplePooling2dTestImpl()
96 queueDescriptor.m_Parameters); in SimplePooling2dTestImpl()
103 queueDescriptor, in SimplePooling2dTestImpl()
|
| /external/armnn/src/backends/aclCommon/ |
| D | ArmComputeUtils.hpp | 103 ConvertAdditionalInfoToAclActivationLayerInfo(const QueueDescriptor& queueDescriptor) in ConvertAdditionalInfoToAclActivationLayerInfo() argument
105 …const ActivationDescriptor* activationDescPtr = queueDescriptor.GetAdditionalInformation<Activatio… in ConvertAdditionalInfoToAclActivationLayerInfo()
310 …rm_compute::Conv3dInfo ComputeConv3DInfo(const armnn::Convolution3dQueueDescriptor queueDescriptor, in ComputeConv3DInfo() argument
313 auto descriptor = queueDescriptor.m_Parameters; in ComputeConv3DInfo()
321 ConvertAdditionalInfoToAclActivationLayerInfo(queueDescriptor); in ComputeConv3DInfo()
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadFactory.cpp | 1705 … auto queueDescriptor = PolymorphicDowncast<const ElementwiseBinaryQueueDescriptor*>(&descriptor); in CreateWorkload() local
1706 return CreateWorkload(LayerType::ElementwiseBinary, *queueDescriptor, info); in CreateWorkload()
|