Home
last modified time | relevance | path

Searched refs:m_DataLayout (Results 1 – 25 of 135) sorted by relevance

123456

/aosp_15_r20/external/armnn/include/armnn/
H A DDescriptors.hpp373 , m_DataLayout(DataLayout::NCHW) in Pooling2dDescriptor()
389 m_DataLayout == rhs.m_DataLayout; in operator ==()
415 DataLayout m_DataLayout; member
437 , m_DataLayout(DataLayout::NCDHW) in Pooling3dDescriptor()
457 m_DataLayout == rhs.m_DataLayout; in operator ==()
491 DataLayout m_DataLayout; member
534 , m_DataLayout(DataLayout::NCHW) in Convolution2dDescriptor()
548 m_DataLayout == rhs.m_DataLayout; in operator ==()
572 DataLayout m_DataLayout; member
592 , m_DataLayout(DataLayout::NDHWC) in Convolution3dDescriptor()
[all …]
/aosp_15_r20/external/armnn/python/pyarmnn/test/
H A Dtest_descriptors.py27 assert desc.m_DataLayout == ann.DataLayout_NCHW
33 assert desc.m_DataLayout == ann.DataLayout_NCHW
47 desc.m_DataLayout = ann.DataLayout_NHWC
48 assert ann.DataLayout_NHWC == desc.m_DataLayout
86 assert desc.m_DataLayout == ann.DataLayout_NCHW
107 assert desc.m_DataLayout == ann.DataLayout_NCHW
124 assert desc.m_DataLayout == ann.DataLayout_NDHWC
130 assert desc.m_DataLayout == ann.DataLayout_NHWC
144 assert desc.m_DataLayout == ann.DataLayout_NCHW
188 assert desc.m_DataLayout == ann.DataLayout_NCHW
[all …]
/aosp_15_r20/external/armnn/python/pyarmnn/src/pyarmnn/swig/modules/
H A Darmnn_descriptors.i106 DataLayout m_DataLayout; member
129 DataLayout m_DataLayout; member
292 DataLayout m_DataLayout; member
335 DataLayout m_DataLayout; member
355 DataLayout m_DataLayout; member
389 DataLayout m_DataLayout; member
531 DataLayout m_DataLayout; member
593 DataLayout m_DataLayout; member
644 DataLayout m_DataLayout; member
756 DataLayout m_DataLayout; member
[all …]
/aosp_15_r20/external/armnn/src/armnn/
H A DSerializeLayerParameters.cpp26 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
71 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
110 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
137 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
187 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
220 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
262 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
346 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
375 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
432 fn("DataLayout", GetDataLayoutName(desc.m_DataLayout)); in Serialize()
[all …]
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClTransposeConvolution2dWorkload.cpp32 …arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
33 …rm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
34 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
43 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClTransposeConvolution2dWorkloadValidate()
84 BuildArmComputeTensor(*m_WeightsTensor, weightInfo, m_Data.m_Parameters.m_DataLayout); in ClTransposeConvolution2dWorkload()
89 …rmComputeTensor(*m_BiasesTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout); in ClTransposeConvolution2dWorkload()
97 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClTransposeConvolution2dWorkload()
H A DClBatchNormalizationFloatWorkload.cpp29 armcomputetensorutils::BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
31 armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
33 armcomputetensorutils::BuildArmComputeTensorInfo(mean, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
35 armcomputetensorutils::BuildArmComputeTensorInfo(var, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
37 armcomputetensorutils::BuildArmComputeTensorInfo(beta, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
39 armcomputetensorutils::BuildArmComputeTensorInfo(gamma, descriptor.m_DataLayout); in ClBatchNormalizationValidate()
83 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClBatchNormalizationFloatWorkload()
H A DClDepthwiseConvolutionWorkload.cpp33 …arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
34 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
47 …Permuted, aclDepthMultiplier) = Convert1HWOTensorInfoToAcl(weights, input,descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
50 …e::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weightsPermuted, descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
64 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClDepthwiseConvolutionWorkloadValidate()
124 … ComputeDepthwiseConv2dDepthMultiplier(m_Data.m_Parameters.m_DataLayout, weightsShape, inputShape); in ClDepthwiseConvolutionWorkload()
136 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClDepthwiseConvolutionWorkload()
H A DClL2NormalizationFloatWorkload.cpp21 …onst arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClL2NormalizationWorkloadValidate()
22 …nst arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClL2NormalizationWorkloadValidate()
24 int axis = (descriptor.m_DataLayout == DataLayout::NCHW) ? 2 : 0; in ClL2NormalizationWorkloadValidate()
45 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClL2NormalizationFloatWorkload()
49 int axis = (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW) ? 2 : 0; in ClL2NormalizationFloatWorkload()
H A DClConvolution3dWorkload.cpp31 …t arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClConvolution3dWorkloadValidate()
32 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in ClConvolution3dWorkloadValidate()
39 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClConvolution3dWorkloadValidate()
43 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClConvolution3dWorkloadValidate()
78 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClConvolution3dWorkload()
H A DClConvolution2dWorkload.cpp31 …t arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClConvolution2dWorkloadValidate()
32 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClConvolution2dWorkloadValidate()
33 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in ClConvolution2dWorkloadValidate()
51 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in ClConvolution2dWorkloadValidate()
102 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClConvolution2dWorkload()
H A DClInstanceNormalizationWorkload.cpp22 … arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in ClInstanceNormalizationWorkloadValidate()
23 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in ClInstanceNormalizationWorkloadValidate()
49 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClInstanceNormalizationWorkload()
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonBatchNormalizationWorkload.cpp34 armcomputetensorutils::BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
36 armcomputetensorutils::BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
38 armcomputetensorutils::BuildArmComputeTensorInfo(mean, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
40 armcomputetensorutils::BuildArmComputeTensorInfo(var, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
42 armcomputetensorutils::BuildArmComputeTensorInfo(beta, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
44 armcomputetensorutils::BuildArmComputeTensorInfo(gamma, descriptor.m_DataLayout); in NeonBatchNormalizationValidate()
74 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonBatchNormalizationWorkload()
H A DNeonTransposeConvolution2dWorkload.cpp31 …arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonTransposeConvolution2dWorkloadValidate()
32 …rm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonTransposeConvolution2dWorkloadValidate()
33 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in NeonTransposeConvolution2dWorkloadValidate()
42 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonTransposeConvolution2dWorkloadValidate()
65 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonTransposeConvolution2dWorkload()
70 …ComputeTensor(*m_KernelTensor, m_Data.m_Weight->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout); in NeonTransposeConvolution2dWorkload()
75 …dArmComputeTensor(*m_BiasTensor, m_Data.m_Bias->GetTensorInfo(), m_Data.m_Parameters.m_DataLayout); in NeonTransposeConvolution2dWorkload()
H A DNeonDepthwiseConvolutionWorkload.cpp36 …arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
37 …rm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
50 …ermuted, aclDepthMultiplier) = Convert1HWOTensorInfoToAcl(weights, input, descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
53 …e::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weightsPermuted, descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
67 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonDepthwiseConvolutionWorkloadValidate()
111 … ComputeDepthwiseConv2dDepthMultiplier(m_Data.m_Parameters.m_DataLayout, weightsShape, inputShape); in NeonDepthwiseConvolutionWorkload()
119 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonDepthwiseConvolutionWorkload()
H A DNeonConvolution2dWorkload.cpp32 …t arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonConvolution2dWorkloadValidate()
33 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonConvolution2dWorkloadValidate()
34 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in NeonConvolution2dWorkloadValidate()
52 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonConvolution2dWorkloadValidate()
88 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonConvolution2dWorkload()
93 …ildArmComputeTensor(*m_KernelTensor, info.m_InputTensorInfos[1], m_Data.m_Parameters.m_DataLayout); in NeonConvolution2dWorkload()
97 …BuildArmComputeTensor(*m_BiasTensor, info.m_InputTensorInfos[2], m_Data.m_Parameters.m_DataLayout); in NeonConvolution2dWorkload()
H A DNeonL2NormalizationFloatWorkload.cpp23 …const arm_compute::TensorInfo aclInput = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonL2NormalizationWorkloadValidate()
24 …nst arm_compute::TensorInfo aclOutput = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonL2NormalizationWorkloadValidate()
26 int axis = (descriptor.m_DataLayout == DataLayout::NCHW) ? 2 : 0; in NeonL2NormalizationWorkloadValidate()
46 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonL2NormalizationFloatWorkload()
50 int axis = (m_Data.m_Parameters.m_DataLayout == DataLayout::NCHW) ? 2 : 0; in NeonL2NormalizationFloatWorkload()
H A DNeonConvolution3dWorkload.cpp32 …t arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonConvolution3dWorkloadValidate()
33 …m_compute::TensorInfo aclWeightsInfo = BuildArmComputeTensorInfo(weights, descriptor.m_DataLayout); in NeonConvolution3dWorkloadValidate()
40 aclBiasesInfo = BuildArmComputeTensorInfo(biases.value(), descriptor.m_DataLayout); in NeonConvolution3dWorkloadValidate()
43 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonConvolution3dWorkloadValidate()
77 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonConvolution3dWorkload()
H A DNeonInstanceNormalizationWorkload.cpp23 … arm_compute::TensorInfo aclInputInfo = BuildArmComputeTensorInfo(input, descriptor.m_DataLayout); in NeonInstanceNormalizationWorkloadValidate()
24 …arm_compute::TensorInfo aclOutputInfo = BuildArmComputeTensorInfo(output, descriptor.m_DataLayout); in NeonInstanceNormalizationWorkloadValidate()
49 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonInstanceNormalizationWorkload()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DResizeTestImpl.cpp28 , m_DataLayout(armnn::DataLayout::NCHW) in ResizeTestParams()
37 armnn::DataLayout m_DataLayout; member
103 if (params.m_DataLayout == armnn::DataLayout::NHWC) in ResizeTestImpl()
114 descriptor.m_Parameters.m_DataLayout = params.m_DataLayout; in ResizeTestImpl()
118 armnnUtils::DataLayoutIndexed dataLayoutIndexed(params.m_DataLayout); in ResizeTestImpl()
160 testParams.m_DataLayout = dataLayout; in ResizeBilinearNopTest()
194 testParams.m_DataLayout = dataLayout; in SimpleResizeBilinearTest()
234 testParams.m_DataLayout = dataLayout; in ResizeBilinearSqMinTest()
275 testParams.m_DataLayout = dataLayout; in ResizeBilinearMinTest()
314 testParams.m_DataLayout = dataLayout; in ResizeBilinearMagTest()
[all …]
H A DPooling3dTestImpl.cpp55 const armnn::DataLayout dataLayout = descriptor.m_DataLayout; in SimplePooling3dTestImpl()
97 queueDescriptor.m_Parameters.m_DataLayout = dataLayout; in SimplePooling3dTestImpl()
166 descriptor.m_DataLayout = dataLayout; in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
280 descriptor.m_DataLayout = dataLayout; in SimpleMaxPooling3dTestCommon()
359 descriptor.m_DataLayout = dataLayout; in IgnorePaddingSimpleMaxPooling3dTestCommon()
438 descriptor.m_DataLayout = dataLayout; in SimpleAveragePooling3dTestCommon()
517 descriptor.m_DataLayout = dataLayout; in LargeTensorsAveragePooling3dTestCommon()
570 descriptor.m_DataLayout = dataLayout; in IgnorePaddingSimpleAveragePooling3dTestCommon()
650 descriptor.m_DataLayout = dataLayout; in SimpleL2Pooling3dTestCommon()
729 descriptor.m_DataLayout = dataLayout; in IgnorePaddingSimpleL2Pooling3dTestCommon()
[all …]
/aosp_15_r20/external/armnn/src/armnn/test/optimizations/
H A DFoldPadTests.cpp46 convolution2dDescriptor.m_DataLayout = DataLayout::NHWC;
73 …(conv2dLayerParams.m_BiasEnabled == false) && (conv2dLayerParams.m_DataLayout == DataLayout::NHWC); in __anonf6df302a0102()
91 …(conv2dLayerParams.m_BiasEnabled == false) && (conv2dLayerParams.m_DataLayout == DataLayout::NHWC); in __anonf6df302a0202()
128 depthwiseConvolution2dDescriptor.m_DataLayout = DataLayout::NHWC;
157 (depthwiseConv2dLayerParams.m_DataLayout == DataLayout::NHWC); in __anonf6df302a0302()
177 (depthwiseConv2dLayerParams.m_DataLayout == DataLayout::NHWC); in __anonf6df302a0402()
214 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC;
299 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC;
370 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC;
443 pooling2dDescriptor.m_DataLayout = DataLayout::NHWC;
[all …]
H A DPermuteAndBatchToSpaceAsDepthToSpaceTests.cpp40 batchToSpaceDesc.m_DataLayout = DataLayout::NHWC; in CreateTestNetworkImpl()
71 batchToSpaceDesc.m_DataLayout = DataLayout::NHWC; in CreateTransposeTestNetworkImpl()
103 … static_cast<const DepthToSpaceLayer*>(layer)->GetParameters().m_DataLayout == DataLayout::NHWC && in __anon772863ce0202()
133 … static_cast<const DepthToSpaceLayer*>(layer)->GetParameters().m_DataLayout == DataLayout::NHWC && in __anon772863ce0302()
167 batchToSpaceDesc.m_DataLayout = DataLayout::NHWC; in CreateTestNetwork()
198 batchToSpaceDesc.m_DataLayout = DataLayout::NHWC; in CreateTransposeTestNetwork()
/aosp_15_r20/external/armnn/src/armnn/test/
H A DTestNameAndDescriptorLayerVisitor.cpp115 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
125 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
152 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
184 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
201 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
265 descriptor.m_DataLayout = armnn::DataLayout::NHWC; in GetDescriptor()
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DTensorBufferArrayView.hpp26 , m_DataLayout(dataLayout) in TensorBufferArrayView()
33 return m_Data[m_DataLayout.GetIndex(m_Shape, b, c, h, w)]; in Get()
39 armnnUtils::DataLayoutIndexed m_DataLayout; member in armnn::TensorBufferArrayView
/aosp_15_r20/external/armnn/src/armnn/optimizations/
H A DPermuteDepthwiseConv2dWeights.hpp47 if (layer->GetParameters().m_DataLayout == armnn::DataLayout::NHWC) in AddPermuteLayer()
52 else if (layer->GetParameters().m_DataLayout == armnn::DataLayout::NCHW) in AddPermuteLayer()
73 … GetDataLayoutName(layer->GetParameters().m_DataLayout))); in AddPermuteLayer()

123456