| /external/armnn/src/armnnUtils/ |
| D | ParserHelper.cpp | 19 void ProcessConcatInputTensorInfo(armnn::TensorInfo& inputTensorInfo, in ProcessConcatInputTensorInfo() argument
28 if (inputTensorInfo.GetNumDimensions() != inputRank) in ProcessConcatInputTensorInfo()
33 inputTensorInfo.GetNumDimensions(), in ProcessConcatInputTensorInfo()
44 mergeDimOrigin += inputTensorInfo.GetShape()[concatAxis]; in ProcessConcatInputTensorInfo()
52 void CalculateReducedOutputTensoInfo(const armnn::TensorInfo& inputTensorInfo, in CalculateReducedOutputTensoInfo() argument
61 for (unsigned int i = 0; i < inputTensorInfo.GetNumDimensions(); ++i) in CalculateReducedOutputTensoInfo()
75 size *= inputTensorInfo.GetShape()[i]; in CalculateReducedOutputTensoInfo()
79 outputShapeVector.push_back(inputTensorInfo.GetShape()[i]); in CalculateReducedOutputTensoInfo()
93 … armnn::TensorShape outputTensorShape(inputTensorInfo.GetNumDimensions(), &outputShapeVector[0]); in CalculateReducedOutputTensoInfo()
94 outputTensorInfo = armnn::TensorInfo(outputTensorShape, inputTensorInfo.GetDataType()); in CalculateReducedOutputTensoInfo()
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/layerTests/ |
| D | PermuteTestImpl.hpp | 24 armnn::TensorInfo inputTensorInfo, in SimplePermuteTestImpl() argument
33 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimplePermuteTestImpl()
39 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); in SimplePermuteTestImpl()
65 armnn::TensorInfo inputTensorInfo; in SimplePermuteTest() local
74 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SimplePermuteTest()
82 inputTensorInfo.SetQuantizationScale(qScale); in SimplePermuteTest()
83 inputTensorInfo.SetQuantizationOffset(qOffset); in SimplePermuteTest()
105 descriptor, inputTensorInfo, in SimplePermuteTest()
115 armnn::TensorInfo inputTensorInfo; in PermuteValueSet1Test() local
124 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in PermuteValueSet1Test()
[all …]
|
| D | TransposeTestImpl.hpp | 25 armnn::TensorInfo inputTensorInfo, in SimpleTransposeTestImpl() argument
33 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimpleTransposeTestImpl()
39 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); in SimpleTransposeTestImpl()
67 armnn::TensorInfo inputTensorInfo; in SimpleTransposeTest() local
76 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SimpleTransposeTest()
84 inputTensorInfo.SetQuantizationScale(qScale); in SimpleTransposeTest()
85 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleTransposeTest()
107 descriptor, inputTensorInfo, in SimpleTransposeTest()
117 armnn::TensorInfo inputTensorInfo; in TransposeValueSet1Test() local
126 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in TransposeValueSet1Test()
[all …]
|
| D | ReduceProdTestImpl.cpp | 22 const armnn::TensorInfo inputTensorInfo, in ReduceTestCommon() argument
31 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReduceTestCommon()
35 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ReduceTestCommon()
45 resolvedAxis = inputTensorInfo.GetNumDimensions() + static_cast<uint32_t>(vAxis[i]); in ReduceTestCommon()
59 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ReduceTestCommon()
92 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSimpleTest() local
96 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceProdSimpleTest()
97 inputTensorInfo.SetQuantizationOffset(0); in ReduceProdSimpleTest()
108 inputTensorInfo, in ReduceProdSimpleTest()
125 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceProdSingleAxisTest1() local
[all …]
|
| D | ReductionTestImpl.cpp | 24 const armnn::TensorInfo inputTensorInfo, in ReductionTestCommon() argument
33 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReductionTestCommon()
37 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ReductionTestCommon()
47 resolvedAxis = inputTensorInfo.GetNumDimensions() + static_cast<uint32_t>(vAxis[i]); in ReductionTestCommon()
61 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ReductionTestCommon()
94 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMaxSimpleTest() local
98 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceMaxSimpleTest()
99 inputTensorInfo.SetQuantizationOffset(0); in ReduceMaxSimpleTest()
117 inputTensorInfo, in ReduceMaxSimpleTest()
134 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceMaxNegativeAxisTest() local
[all …]
|
| D | ReduceSumTestImpl.cpp | 23 const armnn::TensorInfo inputTensorInfo, in ReduceTestCommon() argument
32 auto inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ReduceTestCommon()
36 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ReduceTestCommon()
46 resolvedAxis = inputTensorInfo.GetNumDimensions() + static_cast<uint32_t>(vAxis[i]); in ReduceTestCommon()
60 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ReduceTestCommon()
93 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSimpleTest() local
97 inputTensorInfo.SetQuantizationScale(1.0f); in ReduceSumSimpleTest()
98 inputTensorInfo.SetQuantizationOffset(0); in ReduceSumSimpleTest()
109 inputTensorInfo, in ReduceSumSimpleTest()
126 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ReduceSumSingleAxisTest1() local
[all …]
|
| D | ReshapeTestImpl.cpp | 22 armnn::TensorInfo inputTensorInfo, in SimpleReshapeTestImpl() argument
30 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimpleReshapeTestImpl()
35 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); in SimpleReshapeTestImpl()
63 armnn::TensorInfo inputTensorInfo; in SimpleReshapeTest() local
69 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SimpleReshapeTest()
70 inputTensorInfo.SetQuantizationScale(1.0f); in SimpleReshapeTest()
92 inputTensorInfo); in SimpleReshapeTest()
107 …workloadFactory, memoryManager, tensorHandleFactory, inputTensorInfo, outputTensorInfo, input, out… in SimpleReshapeTest()
116 armnn::TensorInfo inputTensorInfo; in Reshape5dTest() local
122 inputTensorInfo = armnn::TensorInfo(5, inputShape, ArmnnType); in Reshape5dTest()
[all …]
|
| D | ChannelShuffleTestImpl.cpp | 21 armnn::TensorInfo inputTensorInfo, in ChannelShuffleTestImpl() argument
29 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ChannelShuffleTestImpl()
35 AddInputToWorkload(data, info, inputTensorInfo, inputHandle.get()); in ChannelShuffleTestImpl()
63 armnn::TensorInfo inputTensorInfo; in SimpleChannelShuffleTest() local
73 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SimpleChannelShuffleTest()
74 inputTensorInfo.SetQuantizationScale(1.0f); in SimpleChannelShuffleTest()
82 inputTensorInfo); in SimpleChannelShuffleTest()
94 inputTensorInfo, in SimpleChannelShuffleTest()
106 armnn::TensorInfo inputTensorInfo; in ChannelShuffle2DTest() local
116 inputTensorInfo = armnn::TensorInfo(2, inputShape, ArmnnType); in ChannelShuffle2DTest()
[all …]
|
| D | StridedSliceTestImpl.cpp | 25 armnn::TensorInfo& inputTensorInfo, in StridedSliceTestImpl() argument
36 inputTensorInfo.SetQuantizationScale(qScale); in StridedSliceTestImpl()
37 inputTensorInfo.SetQuantizationOffset(qOffset); in StridedSliceTestImpl()
48 tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in StridedSliceTestImpl()
54 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in StridedSliceTestImpl()
82 armnn::TensorInfo inputTensorInfo; in StridedSlice4dTest() local
93 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in StridedSlice4dTest()
112 inputTensorInfo, outputTensorInfo, input, outputExpected, desc); in StridedSlice4dTest()
121 armnn::TensorInfo inputTensorInfo; in StridedSlice4dReverseTest() local
132 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in StridedSlice4dReverseTest()
[all …]
|
| D | ArgMinMaxTestImpl.cpp | 24 const armnn::TensorInfo inputTensorInfo, in ArgMinMaxTestCommon() argument
30 std::vector<T> inputTensor = ConvertToDataType<ArmnnType>(inputData, inputTensorInfo); in ArgMinMaxTestCommon()
33 …e_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in ArgMinMaxTestCommon()
41 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in ArgMinMaxTestCommon()
74 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMaxSimpleTest() local
78 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMaxSimpleTest()
79 inputTensorInfo.SetQuantizationOffset(0); in ArgMaxSimpleTest()
89 inputTensorInfo, outputTensorInfo, in ArgMaxSimpleTest()
102 armnn::TensorInfo inputTensorInfo(inputShape, ArmnnType); in ArgMinSimpleTest() local
106 inputTensorInfo.SetQuantizationScale(1.0f); in ArgMinSimpleTest()
[all …]
|
| D | Pooling2dTestImpl.cpp | 61 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in SimplePooling2dTestImpl() local
70 inputTensorInfo.SetQuantizationScale(qScale); in SimplePooling2dTestImpl()
71 inputTensorInfo.SetQuantizationOffset(qOffset); in SimplePooling2dTestImpl()
79 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimplePooling2dTestImpl()
87 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in SimplePooling2dTestImpl()
94 result.m_Supported = handle.IsPooling2dSupported(inputTensorInfo, in SimplePooling2dTestImpl()
162 armnn::TensorInfo inputTensorInfo({ batchSize, channels, inputHeight, inputWidth }, ArmnnType); in SimpleMaxPooling2dSize3x3Stride2x4TestCommon() local
168 inputTensorInfo.SetQuantizationScale(qScale); in SimpleMaxPooling2dSize3x3Stride2x4TestCommon()
169 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleMaxPooling2dSize3x3Stride2x4TestCommon()
253 input, outputExpected, inputTensorInfo.GetShape(), outputTensorInfo.GetShape()); in SimpleMaxPooling2dSize3x3Stride2x4TestCommon()
[all …]
|
| D | FullyConnectedTestImpl.cpp | 28 armnn::TensorInfo inputTensorInfo, in SimpleFullyConnectedTestImpl() argument
39 …e_ptr<armnn::ITensorHandle> input0Handle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimpleFullyConnectedTestImpl()
48 AddInputToWorkload(data, info, inputTensorInfo, input0Handle.get()); in SimpleFullyConnectedTestImpl()
103 armnn::TensorInfo inputTensorInfo({ 1, inputChannels, inputHeight, inputWidth }, ArmnnType); in FullyConnectedTest() local
104 inputTensorInfo.SetQuantizationScale(0.1f); in FullyConnectedTest()
105 inputTensorInfo.SetQuantizationOffset(63); in FullyConnectedTest()
116 …biasesDesc.SetQuantizationScale(inputTensorInfo.GetQuantizationScale() * weightsDesc.GetQuantizati… in FullyConnectedTest()
126 inputTensorInfo); in FullyConnectedTest()
140 inputTensorInfo, in FullyConnectedTest()
187 armnn::TensorInfo inputTensorInfo; in FullyConnectedLargeTestCommon() local
[all …]
|
| D | Pooling3dTestImpl.cpp | 74 armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( in SimplePooling3dTestImpl() local
83 inputTensorInfo.SetQuantizationScale(qScale); in SimplePooling3dTestImpl()
84 inputTensorInfo.SetQuantizationOffset(qOffset); in SimplePooling3dTestImpl()
92 …ue_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SimplePooling3dTestImpl()
100 AddInputToWorkload(queueDescriptor, workloadInfo, inputTensorInfo, inputHandle.get()); in SimplePooling3dTestImpl()
107 result.m_Supported = handle.IsPooling3dSupported(inputTensorInfo, in SimplePooling3dTestImpl()
183 …armnn::TensorInfo inputTensorInfo = armnnUtils::GetTensorInfo( batchSize, channels, inputDepth, in… in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local
191 inputTensorInfo.SetQuantizationScale(qScale); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
192 inputTensorInfo.SetQuantizationOffset(qOffset); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
257 PermuteNCDHWToNDHWC(input, inputTensorInfo); in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon()
[all …]
|
| D | StackTestImpl.cpp | 28 const armnn::TensorInfo& inputTensorInfo, in StackTestHelper() argument
47 inputHandles.push_back(tensorHandleFactory.CreateTensorHandle(inputTensorInfo)); in StackTestHelper()
53 descriptor.m_Parameters.m_InputShape = inputTensorInfo.GetShape(); in StackTestHelper()
60 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in StackTestHelper()
94 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackAxis0TestImpl() local
149 inputTensorInfo, in StackAxis0TestImpl()
163 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackOutput4DAxis1TestImpl() local
219 inputTensorInfo, in StackOutput4DAxis1TestImpl()
233 armnn::TensorInfo inputTensorInfo ({ 3, 2, 3 }, ArmnnType); in StackOutput4DAxis2TestImpl() local
287 inputTensorInfo, in StackOutput4DAxis2TestImpl()
[all …]
|
| D | SpaceToDepthTestImpl.cpp | 27 armnn::TensorInfo& inputTensorInfo, in SpaceToDepthTestImpl() argument
40 inputTensorInfo = armnnUtils::Permuted(inputTensorInfo, NHWCToNCHW); in SpaceToDepthTestImpl()
44 armnnUtils::Permute(inputTensorInfo.GetShape(), NHWCToNCHW, in SpaceToDepthTestImpl()
56 inputTensorInfo.SetQuantizationScale(qScale); in SpaceToDepthTestImpl()
57 inputTensorInfo.SetQuantizationOffset(qOffset); in SpaceToDepthTestImpl()
66 …e_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo); in SpaceToDepthTestImpl()
70 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in SpaceToDepthTestImpl()
112 armnn::TensorInfo inputTensorInfo; in SpaceToDepthSimpleTest1() local
119 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in SpaceToDepthSimpleTest1()
124 inputTensorInfo, outputTensorInfo, input, outputExpected, desc); in SpaceToDepthSimpleTest1()
[all …]
|
| D | DebugTestImpl.cpp | 27 armnn::TensorInfo& inputTensorInfo, in DebugTestImpl() argument
41 inputTensorInfo.SetQuantizationScale(qScale); in DebugTestImpl()
42 inputTensorInfo.SetQuantizationOffset(qOffset); in DebugTestImpl()
54 …unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.CreateTensorHandle(inputTensorInfo); in DebugTestImpl()
59 AddInputToWorkload(descriptor, info, inputTensorInfo, inputHandle.get()); in DebugTestImpl()
112 armnn::TensorInfo inputTensorInfo; in Debug4dTest() local
124 inputTensorInfo = armnn::TensorInfo(4, inputShape, ArmnnType); in Debug4dTest()
153 inputTensorInfo, in Debug4dTest()
169 armnn::TensorInfo inputTensorInfo; in Debug3dTest() local
181 inputTensorInfo = armnn::TensorInfo(3, inputShape, ArmnnType); in Debug3dTest()
[all …]
|
| /external/armnn/src/backends/backendsCommon/ |
| D | WorkloadData.cpp | 182 const TensorInfo& inputTensorInfo, in ValidateBiasTensorQuantization() argument
222 const float expectedScale = inputTensorInfo.GetQuantizationScale() * weightScales[i]; in ValidateBiasTensorQuantization()
229 …const float expectedScale = inputTensorInfo.GetQuantizationScale() * weightsTensorInfo.GetQuantiza… in ValidateBiasTensorQuantization()
523 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
526 …ValidateTensorNumElementsMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output… in Validate()
527 …ValidateTensorDataTypesMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output"); in Validate()
636 const TensorInfo& inputTensorInfo = workloadInfo.m_InputTensorInfos[0]; in Validate() local
649 ValidateDataTypes(inputTensorInfo, supportedTypes, descriptorName); in Validate()
650 …ValidateTensorDataTypesMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output"); in Validate()
651 ValidateTensorShapesMatch(inputTensorInfo, outputTensorInfo, descriptorName, "input", "output"); in Validate()
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | SpaceToDepthEndToEndTestImpl.cpp | 37 TensorInfo inputTensorInfo(inputShape, DataType, qScale, qOffset, true); in CreateSpaceToDepthNetwork() local
52 Connect(input, SpaceToDepth, inputTensorInfo, 0, 0); in CreateSpaceToDepthNetwork()
63 armnn::TensorInfo& inputTensorInfo, in SpaceToDepthEndToEnd() argument
73 PermuteTensorNhwcToNchw<float>(inputTensorInfo, inputData); in SpaceToDepthEndToEnd()
79 inputTensorInfo.GetShape(), in SpaceToDepthEndToEnd()
105 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest1() local
122 inputTensorInfo, in SpaceToDepthNhwcEndToEndTest1()
136 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNchwEndToEndTest1() local
153 inputTensorInfo, in SpaceToDepthNchwEndToEndTest1()
170 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in SpaceToDepthNhwcEndToEndTest2() local
[all …]
|
| D | FullyConnectedEndToEndTestImpl.hpp | 22 …n::INetworkPtr CreateFullyConnectedNetworkNonConstWeights(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNonConstWeights() argument
34 Connect(inputLayer, fullyConnectedLayer, inputTensorInfo, 0, 0); in CreateFullyConnectedNetworkNonConstWeights()
41 …rkPtr CreateFullyConnectedNetworkNonConstWeightsConstBias(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNonConstWeightsConstBias() argument
56 Connect(inputLayer, fullyConnectedLayer, inputTensorInfo, 0, 0); in CreateFullyConnectedNetworkNonConstWeightsConstBias()
64 …rkPtr CreateFullyConnectedNetworkConstWeightsNonConstBias(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkConstWeightsNonConstBias() argument
79 Connect(inputLayer, fullyConnectedLayer, inputTensorInfo, 0, 0); in CreateFullyConnectedNetworkConstWeightsNonConstBias()
87 …rkPtr CreateFullyConnectedNetworkNoTensorInfoConstWeights(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoTensorInfoConstWeights() argument
99 Connect(inputLayer, fullyConnectedLayer, inputTensorInfo, 0, 0); in CreateFullyConnectedNetworkNoTensorInfoConstWeights()
106 …Ptr CreateFullyConnectedNetworkNoConnectedWeightsExplicit(const armnn::TensorInfo& inputTensorInfo, in CreateFullyConnectedNetworkNoConnectedWeightsExplicit() argument
121 Connect(inputLayer, fullyConnectedLayer, inputTensorInfo, 0, 0); in CreateFullyConnectedNetworkNoConnectedWeightsExplicit()
[all …]
|
| D | InstanceNormalizationEndToEndTestImpl.cpp | 39 TensorInfo inputTensorInfo(inputShape, DataType, qScale, qOffset, true); in CreateInstanceNormalizationNetwork() local
50 Connect(input, instanceNormalization, inputTensorInfo, 0, 0); in CreateInstanceNormalizationNetwork()
61 armnn::TensorInfo& inputTensorInfo, in InstanceNormalizationEndToEnd() argument
73 PermuteTensorNhwcToNchw<float>(inputTensorInfo, inputData); in InstanceNormalizationEndToEnd()
78 …INetworkPtr net = CreateInstanceNormalizationNetwork<DataType::Float32>(inputTensorInfo.GetShape(), in InstanceNormalizationEndToEnd()
107 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNhwcEndToEndTest1() local
159 inputTensorInfo, in InstanceNormalizationNhwcEndToEndTest1()
177 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNchwEndToEndTest1() local
230 inputTensorInfo, in InstanceNormalizationNchwEndToEndTest1()
251 TensorInfo inputTensorInfo(inputShape, DataType::Float32, 0.0f, 0, true); in InstanceNormalizationNhwcEndToEndTest2() local
[all …]
|
| /external/armnn/src/backends/reference/workloads/ |
| D | ArgMinMax.cpp | 16 void ArgMinMax(Decoder<float>& in, OUT* out, const TensorInfo& inputTensorInfo, in ArgMinMax() argument
21 unsigned int uAxis = armnnUtils::GetUnsignedAxis(inputTensorInfo.GetNumDimensions(), axis); in ArgMinMax()
23 …const unsigned int outerElements = armnnUtils::GetNumElementsBetween(inputTensorInfo.GetShape(), 0… in ArgMinMax()
24 const unsigned int axisSize = inputTensorInfo.GetShape()[uAxis]; in ArgMinMax()
25 const unsigned int innerElements = armnnUtils::GetNumElementsBetween(inputTensorInfo.GetShape(), in ArgMinMax()
27 … inputTensorInfo.GetNumDimensions()); in ArgMinMax()
49 template void ArgMinMax(Decoder<float>& in, int32_t* out, const TensorInfo& inputTensorInfo,
52 template void ArgMinMax(Decoder<float>& in, int64_t* out, const TensorInfo& inputTensorInfo,
|
| D | RefCastWorkload.cpp | 44 TensorInfo inputTensorInfo(GetTensorInfo(inputs[0])); in Execute() local
48 if (inputTensorInfo.IsQuantized()) in Execute()
50 inputTensorInfo.SetQuantizationScale(1.0f); in Execute()
51 inputTensorInfo.SetQuantizationOffset(0); in Execute()
59 Cast(*MakeDecoder<float>(inputTensorInfo, inputs[0]->Map()), in Execute()
61 inputTensorInfo.GetNumElements()); in Execute()
|
| /external/armnn/delegate/classic/src/ |
| D | Split.hpp | 45 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitOperator() local
52 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitSplitOperator()
63 const unsigned int splitDim = ComputeWrappedIndex(axis, inputTensorInfo.GetNumDimensions()); in VisitSplitOperator()
77 auto inputDimSize = inputTensorInfo.GetNumDimensions(); in VisitSplitOperator()
92 splitterDimSizes[i] = inputTensorInfo.GetShape()[i]; in VisitSplitOperator()
127 inputTensorInfo, in VisitSplitOperator()
183 const armnn::TensorInfo& inputTensorInfo = GetTensorInfoForTfLiteTensor(tfLiteInputTensor); in VisitSplitVOperator() local
192 auto inputDimensions = static_cast<int32_t>(inputTensorInfo.GetNumDimensions()); in VisitSplitVOperator()
200 …const unsigned int splitDim = ComputeWrappedIndex(axisTensorData[0], inputTensorInfo.GetNumDimensi… in VisitSplitVOperator()
234 auto inputDimSize = inputTensorInfo.GetNumDimensions(); in VisitSplitVOperator()
[all …]
|
| /external/armnn/src/armnnUtils/test/ |
| D | ParserHelperTest.cpp | 24 TensorInfo inputTensorInfo(3, &inputShape[0], DataType::Float32); variable
30 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData1, keepDims, outputTensorInfo1);
39 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData2, keepDims, outputTensorInfo2);
48 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData3, keepDims, outputTensorInfo3);
59 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData4, keepDims, outputTensorInfo4);
72 CalculateReducedOutputTensoInfo(inputTensorInfo, axisData5, keepDims, outputTensorInfo5);
|
| /external/armnn/src/armnnTfLiteParser/test/ |
| D | OutputShapeOfSqueeze.cpp | 29 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32); variable
30 … armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
40 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32); variable
41 … armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
51 … armnn::TensorInfo inputTensorInfo = armnn::TensorInfo(4, m_InputShape, armnn::DataType::Float32); variable
52 … armnn::TensorInfo outputTensorInfo = m_Parser.OutputShapeOfSqueeze(squeezeDims, inputTensorInfo);
|