Home
last modified time | relevance | path

Searched refs:aclDataLayout (Results 1 – 25 of 34) sorted by relevance

12

/external/armnn/src/backends/neon/workloads/
DNeonChannelShuffleWorkload.cpp27 arm_compute::DataLayout aclDataLayout; in NeonChannelShuffleValidate() local
33 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NCHW); in NeonChannelShuffleValidate()
36 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NHWC); in NeonChannelShuffleValidate()
41 aclInputInfo.set_data_layout(aclDataLayout); in NeonChannelShuffleValidate()
42 aclOutputInfo.set_data_layout(aclDataLayout); in NeonChannelShuffleValidate()
69 arm_compute::DataLayout aclDataLayout; in NeonChannelShuffleWorkload() local
73 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NCHW); in NeonChannelShuffleWorkload()
76 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NHWC); in NeonChannelShuffleWorkload()
82 input.info()->set_data_layout(aclDataLayout); in NeonChannelShuffleWorkload()
83 output.info()->set_data_layout(aclDataLayout); in NeonChannelShuffleWorkload()
DNeonResizeWorkload.cpp29 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(descriptor.m_DataLayout); in NeonResizeWorkloadValidate() local
30 aclInputInfo.set_data_layout(aclDataLayout); in NeonResizeWorkloadValidate()
31 aclOutputInfo.set_data_layout(aclDataLayout); in NeonResizeWorkloadValidate()
67 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonResizeWorkload() local
68 input.info()->set_data_layout(aclDataLayout); in NeonResizeWorkload()
69 output.info()->set_data_layout(aclDataLayout); in NeonResizeWorkload()
DNeonConvolution3dWorkload.cpp77 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonConvolution3dWorkload() local
78 input.info()->set_data_layout(aclDataLayout); in NeonConvolution3dWorkload()
79 weights.info()->set_data_layout(aclDataLayout); in NeonConvolution3dWorkload()
80 output.info()->set_data_layout(aclDataLayout); in NeonConvolution3dWorkload()
DNeonInstanceNormalizationWorkload.cpp49 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonInstanceNormalizationWorkload() local
50 input.info()->set_data_layout(aclDataLayout); in NeonInstanceNormalizationWorkload()
51 output.info()->set_data_layout(aclDataLayout); in NeonInstanceNormalizationWorkload()
DNeonPooling2dWorkload.cpp51 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonPooling2dWorkload() local
52 input.info()->set_data_layout(aclDataLayout); in NeonPooling2dWorkload()
53 output.info()->set_data_layout(aclDataLayout); in NeonPooling2dWorkload()
DNeonDepthToSpaceWorkload.cpp44 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonDepthToSpaceWorkload() local
48 input.info()->set_data_layout(aclDataLayout); in NeonDepthToSpaceWorkload()
54 output.info()->set_data_layout(aclDataLayout); in NeonDepthToSpaceWorkload()
DNeonPooling3dWorkload.cpp40 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonPooling3dWorkload() local
41 input.info()->set_data_layout(aclDataLayout); in NeonPooling3dWorkload()
42 output.info()->set_data_layout(aclDataLayout); in NeonPooling3dWorkload()
DNeonSpaceToDepthWorkload.cpp44 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonSpaceToDepthWorkload() local
47 input.info()->set_data_layout(aclDataLayout); in NeonSpaceToDepthWorkload()
52 output.info()->set_data_layout(aclDataLayout); in NeonSpaceToDepthWorkload()
DNeonBatchToSpaceNdWorkload.cpp55 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonBatchToSpaceNdWorkload() local
56 input.info()->set_data_layout(aclDataLayout); in NeonBatchToSpaceNdWorkload()
57 output.info()->set_data_layout(aclDataLayout); in NeonBatchToSpaceNdWorkload()
DNeonDepthwiseConvolutionWorkload.cpp119 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonDepthwiseConvolutionWorkload() local
120 input.info()->set_data_layout(aclDataLayout); in NeonDepthwiseConvolutionWorkload()
121 weights.info()->set_data_layout(aclDataLayout); in NeonDepthwiseConvolutionWorkload()
122 output.info()->set_data_layout(aclDataLayout); in NeonDepthwiseConvolutionWorkload()
DNeonL2NormalizationFloatWorkload.cpp46 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonL2NormalizationFloatWorkload() local
47 input.info()->set_data_layout(aclDataLayout); in NeonL2NormalizationFloatWorkload()
48 output.info()->set_data_layout(aclDataLayout); in NeonL2NormalizationFloatWorkload()
DNeonSpaceToBatchNdWorkload.cpp70 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in NeonSpaceToBatchNdWorkload() local
71 input.info()->set_data_layout(aclDataLayout); in NeonSpaceToBatchNdWorkload()
72 output.info()->set_data_layout(aclDataLayout); in NeonSpaceToBatchNdWorkload()
/external/armnn/src/backends/cl/workloads/
DClChannelShuffleWorkload.cpp30 arm_compute::DataLayout aclDataLayout; in ClChannelShuffleValidate() local
36 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NCHW); in ClChannelShuffleValidate()
39 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NHWC); in ClChannelShuffleValidate()
44 aclInputInfo.set_data_layout(aclDataLayout); in ClChannelShuffleValidate()
45 aclOutputInfo.set_data_layout(aclDataLayout); in ClChannelShuffleValidate()
73 arm_compute::DataLayout aclDataLayout; in ClChannelShuffleWorkload() local
77 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NCHW); in ClChannelShuffleWorkload()
80 aclDataLayout = ConvertDataLayout(armnn::DataLayout::NHWC); in ClChannelShuffleWorkload()
86 input.info()->set_data_layout(aclDataLayout); in ClChannelShuffleWorkload()
87 output.info()->set_data_layout(aclDataLayout); in ClChannelShuffleWorkload()
DClResizeWorkload.cpp29 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(descriptor.m_DataLayout); in ClResizeWorkloadValidate() local
30 aclInputInfo.set_data_layout(aclDataLayout); in ClResizeWorkloadValidate()
31 aclOutputInfo.set_data_layout(aclDataLayout); in ClResizeWorkloadValidate()
65 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClResizeWorkload() local
66 input.info()->set_data_layout(aclDataLayout); in ClResizeWorkload()
67 output.info()->set_data_layout(aclDataLayout); in ClResizeWorkload()
DClConvolution3dWorkload.cpp78 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClConvolution3dWorkload() local
79 input.info()->set_data_layout(aclDataLayout); in ClConvolution3dWorkload()
80 weights.info()->set_data_layout(aclDataLayout); in ClConvolution3dWorkload()
81 output.info()->set_data_layout(aclDataLayout); in ClConvolution3dWorkload()
DClInstanceNormalizationWorkload.cpp49 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClInstanceNormalizationWorkload() local
50 input.info()->set_data_layout(aclDataLayout); in ClInstanceNormalizationWorkload()
51 output.info()->set_data_layout(aclDataLayout); in ClInstanceNormalizationWorkload()
DClPooling2dWorkload.cpp47 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClPooling2dWorkload() local
48 input.info()->set_data_layout(aclDataLayout); in ClPooling2dWorkload()
49 output.info()->set_data_layout(aclDataLayout); in ClPooling2dWorkload()
DClPooling3dWorkload.cpp46 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClPooling3dWorkload() local
47 input.info()->set_data_layout(aclDataLayout); in ClPooling3dWorkload()
48 output.info()->set_data_layout(aclDataLayout); in ClPooling3dWorkload()
DClDepthToSpaceWorkload.cpp52 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClDepthToSpaceWorkload() local
56 input.info()->set_data_layout(aclDataLayout); in ClDepthToSpaceWorkload()
62 output.info()->set_data_layout(aclDataLayout); in ClDepthToSpaceWorkload()
DClSpaceToDepthWorkload.cpp32 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClSpaceToDepthWorkload() local
35 input.info()->set_data_layout(aclDataLayout); in ClSpaceToDepthWorkload()
40 output.info()->set_data_layout(aclDataLayout); in ClSpaceToDepthWorkload()
DClBatchToSpaceNdWorkload.cpp53 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClBatchToSpaceNdWorkload() local
56 input.info()->set_data_layout(aclDataLayout); in ClBatchToSpaceNdWorkload()
63 output.info()->set_data_layout(aclDataLayout); in ClBatchToSpaceNdWorkload()
DClNormalizationFloatWorkload.cpp47 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClNormalizationFloatWorkload() local
48 input.info()->set_data_layout(aclDataLayout); in ClNormalizationFloatWorkload()
49 output.info()->set_data_layout(aclDataLayout); in ClNormalizationFloatWorkload()
DClL2NormalizationFloatWorkload.cpp45 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClL2NormalizationFloatWorkload() local
46 input.info()->set_data_layout(aclDataLayout); in ClL2NormalizationFloatWorkload()
47 output.info()->set_data_layout(aclDataLayout); in ClL2NormalizationFloatWorkload()
DClDepthwiseConvolutionWorkload.cpp136 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClDepthwiseConvolutionWorkload() local
137 input.info()->set_data_layout(aclDataLayout); in ClDepthwiseConvolutionWorkload()
138 weights.info()->set_data_layout(aclDataLayout); in ClDepthwiseConvolutionWorkload()
139 output.info()->set_data_layout(aclDataLayout); in ClDepthwiseConvolutionWorkload()
DClSpaceToBatchNdWorkload.cpp75 arm_compute::DataLayout aclDataLayout = ConvertDataLayout(m_Data.m_Parameters.m_DataLayout); in ClSpaceToBatchNdWorkload() local
76 input.info()->set_data_layout(aclDataLayout); in ClSpaceToBatchNdWorkload()
77 output.info()->set_data_layout(aclDataLayout); in ClSpaceToBatchNdWorkload()

12