Home
last modified time | relevance | path

Searched defs:dataLayout (Results 1 – 25 of 63) sorted by relevance

123

/external/armnn/src/backends/backendsCommon/test/layerTests/
DResizeTestImpl.cpp155 const armnn::DataLayout dataLayout) in ResizeBilinearNopTest()
189 const armnn::DataLayout dataLayout) in SimpleResizeBilinearTest()
229 const armnn::DataLayout dataLayout) in ResizeBilinearSqMinTest()
270 const armnn::DataLayout dataLayout) in ResizeBilinearMinTest()
309 const armnn::DataLayout dataLayout) in ResizeBilinearMagTest()
354 const armnn::DataLayout dataLayout) in ResizeNearestNeighborNopTest()
388 const armnn::DataLayout dataLayout) in SimpleResizeNearestNeighborTest()
428 const armnn::DataLayout dataLayout) in ResizeNearestNeighborSqMinTest()
469 const armnn::DataLayout dataLayout) in ResizeNearestNeighborMinTest()
508 const armnn::DataLayout dataLayout, in ResizeNearestNeighborMagTest()
[all …]
DInstanceNormalizationTestImpl.cpp74 armnn::DataLayout dataLayout) in InstanceNormTest()
159 armnn::DataLayout dataLayout) in InstanceNormTest2()
246 armnn::DataLayout dataLayout) in InstanceNormFloat32Test()
255 armnn::DataLayout dataLayout) in InstanceNormFloat16Test()
264 armnn::DataLayout dataLayout) in InstanceNormFloat32Test2()
273 armnn::DataLayout dataLayout) in InstanceNormFloat16Test2()
DDepthToSpaceTestImpl.cpp82 armnn::DataLayout dataLayout) in DepthToSpaceTest1()
111 armnn::DataLayout dataLayout) in DepthToSpaceTest2()
161 armnn::DataLayout dataLayout) in DepthToSpaceTest3()
188 armnn::DataLayout dataLayout) in DepthToSpaceTest4()
DPooling2dTestImpl.cpp44 const armnn::DataLayout dataLayout = descriptor.m_DataLayout; in SimplePooling2dTestImpl() local
256 const armnn::DataLayout dataLayout = armnn::DataLayout::NCHW, in SimpleMaxPooling2dTestCommon()
328 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW, in SimpleAveragePooling2dTestCommon()
452 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW, in SimpleL2Pooling2dTestCommon()
1443 const armnn::DataLayout dataLayout) in SimpleMaxPooling2dTest()
1453 const armnn::DataLayout dataLayout) in SimpleMaxPooling2dUint8Test()
1463 const armnn::DataLayout dataLayout) in SimpleMaxPooling2dInt16Test()
1526 const armnn::DataLayout dataLayout) in SimpleAveragePooling2dTest()
1536 const armnn::DataLayout dataLayout) in SimpleAveragePooling2dUint8Test()
1546 const armnn::DataLayout dataLayout) in SimpleAveragePooling2dInt16Test()
[all …]
DSpaceToBatchNdTestImpl.cpp94 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW) in SpaceToBatchNdSimpleTest()
130 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW) in SpaceToBatchNdMultiChannelsTest()
171 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW) in SpaceToBatchNdMultiBlockTest()
213 armnn::DataLayout dataLayout = armnn::DataLayout::NCHW) in SpaceToBatchNdPaddingTest()
/external/armnn/src/armnnUtils/
DDataLayoutIndexed.cpp13 DataLayoutIndexed::DataLayoutIndexed(armnn::DataLayout dataLayout) in DataLayoutIndexed()
34 bool operator==(const DataLayout& dataLayout, const DataLayoutIndexed& indexed) in operator ==()
39 bool operator==(const DataLayoutIndexed& indexed, const DataLayout& dataLayout) in operator ==()
DTensorUtils.cpp23 const DataLayout dataLayout) in GetTensorShape()
42 const DataLayout dataLayout, in GetTensorInfo()
/external/armnn/src/backends/backendsCommon/
DWorkloadUtils.cpp37 void ReshapeWeightsForAcl(TensorInfo& weightInfo, DataLayout dataLayout) in ReshapeWeightsForAcl()
63 ConstTensor ReorderWeightChannelsForAcl(const ConstTensor& weightHandle, DataLayout dataLayout, voi… in ReorderWeightChannelsForAcl()
110 …nsorInfo ConvertWeightTensorInfoFromArmnnToAcl(const TensorInfo& weightInfo, DataLayout dataLayout) in ConvertWeightTensorInfoFromArmnnToAcl()
134 DataLayout dataLayout, in ConvertWeightTensorFromArmnnToAcl()
/external/armnn/src/backends/backendsCommon/test/
DResizeEndToEndTestImpl.hpp44 armnn::DataLayout dataLayout, in ResizeEndToEnd()
138 armnn::DataLayout dataLayout) in ResizeBilinearEndToEnd()
145 armnn::DataLayout dataLayout) in ResizeNearestNeighborEndToEnd()
DBatchToSpaceNdEndToEndTestImpl.hpp25 armnn::DataLayout dataLayout, in CreateBatchToSpaceNdNetwork()
50 void BatchToSpaceNdEndToEnd(const std::vector<BackendId>& backends, armnn::DataLayout dataLayout) in BatchToSpaceNdEndToEnd()
79 …BatchToSpaceNdComplexEndToEnd(const std::vector<BackendId>& backends, armnn::DataLayout dataLayout) in BatchToSpaceNdComplexEndToEnd()
DSpaceToDepthEndToEndTestImpl.cpp27 const armnn::DataLayout dataLayout, in CreateSpaceToDepthNetwork()
62 const armnn::DataLayout& dataLayout, in SpaceToDepthEndToEnd()
DInstanceNormalizationEndToEndTestImpl.cpp27 const armnn::DataLayout dataLayout, in CreateInstanceNormalizationNetwork()
60 const armnn::DataLayout& dataLayout, in InstanceNormalizationEndToEnd()
/external/armnn/src/backends/aclCommon/
DArmComputeTensorUtils.cpp134 armnn::DataLayout dataLayout) in BuildArmComputeTensorInfo()
142 arm_compute::DataLayout ConvertDataLayout(armnn::DataLayout dataLayout) in ConvertDataLayout()
168 const DataLayout dataLayout = ConvertDataLayout(descriptor.m_DataLayout); in BuildArmComputePoolingLayerInfo() local
/external/armnn/src/backends/reference/workloads/
DSpaceToDepth.cpp18 const DataLayoutIndexed& dataLayout) in GetOffset()
42 DataLayoutIndexed dataLayout = params.m_DataLayout; in SpaceToDepth() local
DBatchToSpaceNd.cpp20 unsigned int channels, const DataLayoutIndexed& dataLayout) in Offset()
35 void BatchToSpaceNd(const DataLayoutIndexed& dataLayout, in BatchToSpaceNd()
DSpaceToBatchNd.cpp20 const DataLayoutIndexed& dataLayout) in GetOffset()
40 DataLayoutIndexed dataLayout = params.m_DataLayout; in SpaceToBatchNd() local
DTensorBufferArrayView.hpp23 armnnUtils::DataLayoutIndexed dataLayout = DataLayout::NCHW) in TensorBufferArrayView()
/external/armnn/src/backends/neon/workloads/
DNeonDepthToSpaceWorkload.cpp23 DataLayout dataLayout = descriptor.m_DataLayout; in NeonDepthToSpaceWorkloadValidate() local
DNeonSpaceToDepthWorkload.cpp23 DataLayout dataLayout = descriptor.m_DataLayout; in NeonSpaceToDepthWorkloadValidate() local
/external/armnn/src/armnn/optimizations/
DFoldPadIntoConvolution2d.hpp39 armnn::DataLayout dataLayout = descriptor.m_DataLayout; in Run() local
/external/armnn/src/backends/cl/workloads/
DClDepthToSpaceWorkload.cpp26 DataLayout dataLayout = desc.m_DataLayout; in ClDepthToSpaceWorkloadValidate() local
DClSpaceToDepthWorkload.cpp49 DataLayout dataLayout = desc.m_DataLayout; in ClSpaceToDepthWorkloadValidate() local
DClBatchToSpaceNdWorkload.cpp50 DataLayout dataLayout = desc.m_DataLayout; in ClBatchToSpaceNdWorkloadValidate() local
/external/armnn/src/backends/neon/test/
DNeonCreateWorkloadTests.cpp196 static void NeonCreateBatchNormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateBatchNormalizationWorkloadTest()
240 static void NeonCreateConvolution2dWorkloadTest(DataLayout dataLayout = DataLayout::NCHW) in NeonCreateConvolution2dWorkloadTest()
308 static void NeonCreateDepthWiseConvolutionWorkloadTest(DataLayout dataLayout) in NeonCreateDepthWiseConvolutionWorkloadTest()
387 static void NeonCreateNormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateNormalizationWorkloadTest()
431 static void NeonCreatePooling2dWorkloadTest(DataLayout dataLayout = DataLayout::NCHW) in NeonCreatePooling2dWorkloadTest()
555 static void NeonCreateResizeWorkloadTest(DataLayout dataLayout) in NeonCreateResizeWorkloadTest()
799 static void NeonCreateL2NormalizationWorkloadTest(DataLayout dataLayout) in NeonCreateL2NormalizationWorkloadTest()
/external/armnn/src/armnn/test/
DCreateWorkload.hpp335 armnn::IWorkloadFactory& factory, armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW) in CreateBatchNormalizationWorkloadTest()
393 armnn::IWorkloadFactory& factory, armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW) in CreateBatchNormalizationWithBlobWorkloadTest()
475DataLayout dataLayout = DataLayout::NCHW, in CreateConvolution2dWorkloadTest()
537 DataLayout dataLayout = DataLayout::NCHW, in CreateConvolution2dFusedActivationWithBlobWorkloadTest()
622DataLayout dataLayout = DataLayout::NCHW, in CreateConvolution2dWorkloadFastMathTest()
1136 armnn::IWorkloadFactory& factory, armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW) in CreateDepthwiseConvolution2dWorkloadTest()
1305DataLayout dataLayout = DataLayout::NCHW) in CreateNormalizationWorkloadTest()
1357DataLayout dataLayout = DataLayout::NCHW) in CreatePooling2dWorkloadTest()
1654DataLayout dataLayout = DataLayout::NCHW) in CreateResizeBilinearWorkloadTest()
1768 armnn::Graph& graph, DataLayout dataLayout = DataLayout::NCHW) in CreateL2NormalizationWorkloadTest()

123