Home
last modified time | relevance | path

Searched defs:inputTensorInfo (Results 1 – 25 of 164) sorted by relevance

1234567

/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp182 const TensorInfo& inputTensorInfo, in ValidateBiasTensorQuantization()
523 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
636 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
661 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
727 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
754 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
791 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
932 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[i]; in Validate() local
1043 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
1072 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
[all …]
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DStridedSliceTestImpl.cpp25 armnn::TensorInfo& inputTensorInfo, in StridedSliceTestImpl()
82 armnn::TensorInfo inputTensorInfo; in StridedSlice4dTest() local
121 armnn::TensorInfo inputTensorInfo; in StridedSlice4dReverseTest() local
160 armnn::TensorInfo inputTensorInfo; in StridedSliceSimpleStrideTest() local
201 armnn::TensorInfo inputTensorInfo; in StridedSliceSimpleRangeMaskTest() local
246 armnn::TensorInfo inputTensorInfo; in StridedSliceShrinkAxisMaskTest() local
287 armnn::TensorInfo inputTensorInfo; in StridedSliceShrinkAxisMaskBitPosition0Test() local
328 armnn::TensorInfo inputTensorInfo; in StridedSliceShrinkAxisMaskBitPosition1Test() local
369 armnn::TensorInfo inputTensorInfo; in StridedSliceShrinkAxisMaskBitPosition2Test() local
410 armnn::TensorInfo inputTensorInfo; in StridedSliceShrinkAxisMaskBitPosition3Test() local
[all …]
H A DStackTestImpl.cpp28 const armnn::TensorInfo& inputTensorInfo, in StackTestHelper()
94 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackAxis0TestImpl() local
163 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackOutput4DAxis1TestImpl() local
233 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackOutput4DAxis2TestImpl() local
301 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackOutput4DAxis3TestImpl() local
377 armnn::TensorInfo inputTensorInfo ({ 3, 3 }, ArmnnType); in StackOutput3DInputs3TestImpl() local
436 armnn::TensorInfo inputTensorInfo ({ 2, 2, 2, 3 }, ArmnnType); in StackOutput5DTestImpl() local
576 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, armnn::DataType::Float16); in StackFloat16Test() local
H A DPooling2dTestImpl.cpp61 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in SimplePooling2dTestImpl() local
162 armnn::TensorInfo inputTensorInfo({ batchSize, channels, inputHeight, inputWidth }, ArmnnType); in SimpleMaxPooling2dSize3x3Stride2x4TestCommon() local
272 … armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 2, 4, 4, dataLayout, ArmnnType); in SimpleMaxPooling2dTestCommon() local
341 … armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 2, 4, 4, dataLayout, ArmnnType); in SimpleAveragePooling2dTestCommon() local
412 armnn::TensorInfo inputTensorInfo({ 5, 3, 52, 60 }, ArmnnType); in LargeTensorsAveragePooling2dTestCommon() local
459 … armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 2, 4, 4, dataLayout, ArmnnType); in SimpleL2Pooling2dTestCommon() local
517 armnn::TensorInfo inputTensorInfo({ 1, 1, 4, 4 }, ArmnnType); in L2Pooling2dSize3Stride1TestCommon() local
554 armnn::TensorInfo inputTensorInfo({ 1, 1, 9, 9 }, ArmnnType); in L2Pooling2dSize3Stride3TestCommon() local
597 armnn::TensorInfo inputTensorInfo({ 1, 1, 7, 7 }, ArmnnType); in L2Pooling2dSize3Stride4TestCommon() local
637 armnn::TensorInfo inputTensorInfo({ 1, 1, 7, 7 }, ArmnnType); in L2Pooling2dSize7TestCommon() local
[all …]
H A DMirrorPadTestImpl.cpp24 const armnn::TensorInfo& inputTensorInfo, in MirrorPad2dTestCommon()
70 const armnn::TensorInfo& inputTensorInfo, in MirrorPad3dTestCommon()
115 const armnn::TensorInfo& inputTensorInfo, in MirrorPad4dTestCommon()
167 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadSymmetric2dTest() local
218 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadReflect2dTest() local
269 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadSymmetric3dTest() local
336 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadReflect3dTest() local
403 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadSymmetric4dTest() local
716 const armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType, qScale, qOffset); in PadReflect4dTest() local
853 const armnn::TensorInfo inputTensorInfo(inputShape, armnn::DataType::Float16); in PadSymmetricFloat16() local
[all …]
H A DArgMinMaxTestImpl.cpp24 const armnn::TensorInfo inputTensorInfo, in ArgMinMaxTestCommon()
74 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxSimpleTest() local
102 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinSimpleTest() local
130 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinChannelTest() local
166 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxChannelTest() local
202 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxHeightTest() local
238 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinWidthTest() local
H A DReduceSumTestImpl.cpp23 const armnn::TensorInfo inputTensorInfo, in ReduceTestCommon()
93 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSimpleTest() local
126 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest1() local
167 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest2() local
226 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest3() local
291 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumMultipleAxisTest() local
H A DReduceProdTestImpl.cpp22 const armnn::TensorInfo inputTensorInfo, in ReduceTestCommon()
92 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSimpleTest() local
125 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSingleAxisTest1() local
162 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSingleAxisTest2() local
221 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSingleAxisTest3() local
286 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdMultipleAxisTest() local
H A DReductionTestImpl.cpp24 const armnn::TensorInfo inputTensorInfo, in ReductionTestCommon()
94 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMaxSimpleTest() local
134 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMaxNegativeAxisTest() local
175 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMaxSimpleTest2() local
217 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMinSimpleTest() local
257 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMinNegativeAxisTest() local
H A DRankTestImpl.cpp16 armnn::TensorInfo inputTensorInfo, in RankTest()
61 armnn::TensorInfo inputTensorInfo({6}, ArmnnType, 1.0f, 0); in RankDimSize1Test() local
73 armnn::TensorInfo inputTensorInfo({1, 3}, ArmnnType, 1.0f, 0); in RankDimSize2Test() local
85 armnn::TensorInfo inputTensorInfo({1, 3, 2}, ArmnnType, 1.0f, 0); in RankDimSize3Test() local
97 armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, ArmnnType, 1.0f, 0); in RankDimSize4Test() local
H A DPooling3dTestImpl.cpp74 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in SimplePooling3dTestImpl() local
183 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( batchSize, channels, inputDepth, in… in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local
282 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleMaxPooling3dTestCommon() local
361 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 4, 4, 4 , dataLayout, ArmnnTy… in IgnorePaddingSimpleMaxPooling3dTestCommon() local
440 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleAveragePooling3dTestCommon() local
519 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 5, 3, 52, 60, 68, dataLayout, Armnn… in LargeTensorsAveragePooling3dTestCommon() local
573 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo ( 1, 1, 4, 4, 4, dataLayout, ArmnnTy… in IgnorePaddingSimpleAveragePooling3dTestCommon() local
652 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo(1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in SimpleL2Pooling3dTestCommon() local
731 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 4, 4, 4, dataLayout, ArmnnTyp… in IgnorePaddingSimpleL2Pooling3dTestCommon() local
839 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( 1, 1, 1, 3, 1, dataLayout, ArmnnTyp… in AsymmetricNonSquareMaxPooling3dWithPaddingOnlyPoolTestCommon() local
[all …]
H A DShapeTestImpl.cpp16 armnn::TensorInfo inputTensorInfo, in ShapeTest()
59 armnn::TensorInfo inputTensorInfo({ 1 }, ArmnnType, 1.0f, 0); in ShapeDimSize1Test() local
74 armnn::TensorInfo inputTensorInfo({ 1, 2 }, ArmnnType, 1.0f, 0); in ShapeDimSize2Test() local
89 armnn::TensorInfo inputTensorInfo({ 1, 2, 3 }, ArmnnType, 1.0f, 0); in ShapeDimSize3Test() local
104 armnn::TensorInfo inputTensorInfo({ 2, 2, 3, 2 }, ArmnnType, 1.0f, 0); in ShapeDimSize4Test() local
H A DPermuteTestImpl.hpp24 armnn::TensorInfo inputTensorInfo, in SimplePermuteTestImpl()
65 armnn::TensorInfo inputTensorInfo; in SimplePermuteTest() local
115 armnn::TensorInfo inputTensorInfo; in PermuteValueSet1Test() local
166 armnn::TensorInfo inputTensorInfo; in PermuteValueSet2Test() local
217 armnn::TensorInfo inputTensorInfo; in PermuteValueSet3Test() local
H A DTransposeTestImpl.hpp25 armnn::TensorInfo inputTensorInfo, in SimpleTransposeTestImpl()
67 armnn::TensorInfo inputTensorInfo; in SimpleTransposeTest() local
117 armnn::TensorInfo inputTensorInfo; in TransposeValueSet1Test() local
168 armnn::TensorInfo inputTensorInfo; in TransposeValueSet2Test() local
219 armnn::TensorInfo inputTensorInfo; in TransposeValueSet3Test() local
H A DReshapeTestImpl.cpp22 armnn::TensorInfo inputTensorInfo, in SimpleReshapeTestImpl()
63 armnn::TensorInfo inputTensorInfo; in SimpleReshapeTest() local
116 armnn::TensorInfo inputTensorInfo; in Reshape5dTest() local
178 armnn::TensorInfo inputTensorInfo; in ReshapeBooleanTest() local
H A DChannelShuffleTestImpl.cpp21 armnn::TensorInfo inputTensorInfo, in ChannelShuffleTestImpl()
63 armnn::TensorInfo inputTensorInfo; in SimpleChannelShuffleTest() local
106 armnn::TensorInfo inputTensorInfo; in ChannelShuffle2DTest() local
154 armnn::TensorInfo inputTensorInfo; in ChannelShuffle4DTest() local
H A DConv2dTestImpl.cpp243 armnn::TensorInfo inputTensorInfo = in SimpleConvolution2dTestImpl() local
414 … armnn::TensorInfo inputTensorInfo({inputNum, inputHeight, inputWidth, inputChannels}, ArmnnType); in SimpleConvolution2dNhwcTestImpl() local
1070 armnn::TensorInfo& inputTensorInfo, in Convolution2d3x3DilationTestCommon()
1162 armnn::TensorInfo inputTensorInfo({ 1, 1, 10, 10 }, ArmnnType); in Convolution2d3x3Dilation3x3Test() local
1220 armnn::TensorInfo inputTensorInfo({ 1, 2, 10, 10 }, ArmnnType); in Convolution2d2x3x3Dilation3x3Test() local
1293 armnn::TensorInfo inputTensorInfo({ 1, 1, 10, 10 }, ArmnnType); in Convolution2d2x2Dilation2x2Padding2x2Stride3x3Test() local
1380 armnn::TensorInfo inputTensorInfo; in CompareConvolution2dTestImpl() local
1728 armnn::TensorInfo inputTensorInfo = in DepthwiseConvolution2dAsymmetricTestImpl() local
1873 armnn::TensorInfo inputTensorInfo = in DepthwiseConvolution2dDepthMul1TestImpl() local
2046 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in DepthwiseConvolution2dTestImpl() local
[all …]
H A DLogSoftmaxTestImpl.cpp85 armnn::TensorInfo inputTensorInfo(inputOutputShape, ArmnnType); in LogSoftmaxTest1() local
123 armnn::TensorInfo inputTensorInfo(inputOutputShape, ArmnnType); in LogSoftmaxTest2() local
161 armnn::TensorInfo inputTensorInfo(inputOutputShape, ArmnnType); in LogSoftmaxTest3() local
199 armnn::TensorInfo inputTensorInfo(inputOutputShape, ArmnnType); in LogSoftmaxTest4() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DFullyConnectedEndToEndTestImpl.hpp22 …n::INetworkPtr CreateFullyConnectedNetworkNonConstWeights(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNonConstWeights()
41 …rkPtr CreateFullyConnectedNetworkNonConstWeightsConstBias(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNonConstWeightsConstBias()
64 …rkPtr CreateFullyConnectedNetworkConstWeightsNonConstBias(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkConstWeightsNonConstBias()
87 …rkPtr CreateFullyConnectedNetworkNoTensorInfoConstWeights(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoTensorInfoConstWeights()
106 …Ptr CreateFullyConnectedNetworkNoConnectedWeightsExplicit(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoConnectedWeightsExplicit()
128 …kPtr CreateFullyConnectedNetworkNoConnectedWeightsAndBias(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoConnectedWeightsAndBias()
144 …orkPtr CreateFullyConnectedNetworkNoConnectedBiasExplicit(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoConnectedBiasExplicit()
169 armnn::TensorInfo inputTensorInfo({ 1, 1, 2, 3 }, ArmnnType); in FullyConnectedWithDynamicWeightsEndToEnd() local
244 …armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, inputShape, armnn::DataType::Float32, 0.0… in FullyConnectedWithDynamicOrConstantInputsEndToEnd() local
360 …armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, inputShape, armnn::DataType::Float32, 0.0… in FullyConnectedErrorChecking() local
H A DSpaceToDepthEndToEndTestImpl.cpp37 TensorInfo inputTensorInfo(inputShape, DataType, qScale, qOffset, true); in CreateSpaceToDepthNetwork() local
63 armnn::TensorInfo& inputTensorInfo, in SpaceToDepthEndToEnd()
105 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest1() local
136 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNchwEndToEndTest1() local
170 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest2() local
200 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNchwEndToEndTest2() local
H A DWorkloadDataValidation.cpp27 TensorInfo inputTensorInfo(inputShape, armnn::DataType::QAsymmU8, .1f, 125); variable
59 armnn::TensorInfo inputTensorInfo; variable
80 armnn::TensorInfo inputTensorInfo; variable
111 armnn::TensorInfo inputTensorInfo; variable
143 armnn::TensorInfo inputTensorInfo; variable
187 armnn::TensorInfo inputTensorInfo; variable
233 armnn::TensorInfo inputTensorInfo; variable
290 armnn::TensorInfo inputTensorInfo; variable
472 armnn::TensorInfo inputTensorInfo; variable
505 armnn::TensorInfo inputTensorInfo({batchSize , inputSize}, dataType, qScale, qOffset ); variable
H A DInstanceNormalizationEndToEndTestImpl.cpp39 TensorInfo inputTensorInfo(inputShape, DataType, qScale, qOffset, true); in CreateInstanceNormalizationNetwork() local
61 armnn::TensorInfo& inputTensorInfo, in InstanceNormalizationEndToEnd()
107 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNhwcEndToEndTest1() local
177 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNchwEndToEndTest1() local
251 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNhwcEndToEndTest2() local
322 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNchwEndToEndTest2() local
/aosp_15_r20/external/armnn/src/armnnTfLiteParser/
H A DTfLiteParser.cpp874 auto &inputTensorInfo = m_TensorInfos[inputId]; in OutputTensorInfoFromInputs() local
880 … auto inputTensorInfo = ::armnnTfLiteParser::ToTensorInfo(subgraphPtr->tensors[inputId].get()); in OutputTensorInfoFromInputs() local
1234 armnn::TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseConv2D() local
1319 armnn::TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseConv3D() local
1399 armnn::TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseDepthwiseConv2D() local
1485 armnn::TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseExpandDims() local
1569 TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseTranspose() local
1638 armnn::TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 2); in ParseTransposeConv() local
1778 TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseBatchToSpaceND() local
1838 TensorInfo inputTensorInfo = InputTensorInfo(subgraphIndex, operatorIndex, 0); in ParseMaximum() local
[all …]
/aosp_15_r20/external/armnn/src/armnnTestUtils/
H A DCreateWorkload.hpp1440 armnn::TensorInfo inputTensorInfo(inputShape, DataType); in CreateNormalizationWorkloadTest() local
1631 armnn::TensorInfo inputTensorInfo({ 1, 2, 100, 10 }, DataType); in CreateSplitterConcatWorkloadTest() local
1705 armnn::TensorInfo inputTensorInfo ({ 1, 3, 100, 50 }, DataType); in CreateSplitterMultipleInputsOneOutputWorkloadTest() local
1799 armnn::TensorInfo inputTensorInfo(inputShape, DataType); in CreateResizeBilinearWorkloadTest() local
1901 armnn::TensorInfo inputTensorInfo(inputShape, DataType); in CreateL2NormalizationWorkloadTest() local
1934 armnn::TensorInfo inputTensorInfo({ 4, 1 }, DataType); in CreateReshapeWorkloadTest() local
1963 armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, armnn::DataType::Float16); in CreateConvertFp16ToFp32WorkloadTest() local
1992 armnn::TensorInfo inputTensorInfo({1, 3, 2, 3}, armnn::DataType::Float32); in CreateConvertFp32ToFp16WorkloadTest() local
2023 armnn::TensorInfo inputTensorInfo({ 1, 3, 7, 4 }, DataType); in CreateMeanWorkloadTest() local
2048 armnn::TensorInfo inputTensorInfo({ 2, 3, 2, 5 }, DataType); in CreateConcatWorkloadTest() local
[all …]
/aosp_15_r20/external/armnn/src/backends/cl/test/
H A DClDefaultAllocatorTests.cpp32 TensorInfo inputTensorInfo(TensorShape({1, 1}), DataType::Float32); variable
58 TensorInfo inputTensorInfo(TensorShape({2, 1}), DataType::Float32); variable
133 TensorInfo inputTensorInfo(TensorShape({1, 1}), DataType::Float32); variable
159 TensorInfo inputTensorInfo(TensorShape({2, 1}), DataType::Float32); variable

1234567