| /external/armnn/src/armnnSerializer/test/ |
| D | SerializerTests.cpp | 44 inputLayer0->GetOutputSlot(0).SetTensorInfo(tensorInfo);
45 inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);
46 additionLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
74 inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo); in SerializeArgMinMaxTest()
75 argMinMaxLayer->GetOutputSlot(0).SetTensorInfo(outputInfo); in SerializeArgMinMaxTest()
124 inputXLayer->GetOutputSlot(0).SetTensorInfo(inputXInfo);
125 inputYLayer->GetOutputSlot(0).SetTensorInfo(inputYInfo);
126 batchMatMulLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
178 inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
179 batchNormalizationLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
[all …]
|
| D | LstmSerializationTests.cpp | 260 inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
263 outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
266 cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
269 lstmLayer->GetOutputSlot(0).SetTensorInfo(lstmTensorInfoScratchBuff);
272 lstmLayer->GetOutputSlot(1).SetTensorInfo(outputStateTensorInfo);
275 lstmLayer->GetOutputSlot(2).SetTensorInfo(cellStateTensorInfo);
278 lstmLayer->GetOutputSlot(3).SetTensorInfo(outputStateTensorInfo);
406 inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
409 outputStateIn->GetOutputSlot(0).SetTensorInfo(outputStateTensorInfo);
412 cellStateIn->GetOutputSlot(0).SetTensorInfo(cellStateTensorInfo);
[all …]
|
| /external/armnn/src/armnn/test/optimizations/ |
| D | MoveTransposeUpTests.cpp | 34 head->GetOutputHandler().SetTensorInfo(transposed);
38 head->GetOutputHandler().SetTensorInfo(info);
41 head->GetOutputHandler().SetTensorInfo(info);
46 .SetTensorInfo(info);
50 head->GetOutputHandler().SetTensorInfo(info);
53 head->GetOutputHandler().SetTensorInfo(info);
56 head->GetOutputHandler().SetTensorInfo(info);
59 head->GetOutputHandler().SetTensorInfo(info);
64 .SetTensorInfo(info);
69 .SetTensorInfo(info);
|
| D | MovePermuteUpTests.cpp | 33 head->GetOutputHandler().SetTensorInfo(permuted);
37 head->GetOutputHandler().SetTensorInfo(info);
40 head->GetOutputHandler().SetTensorInfo(info);
45 .SetTensorInfo(info);
49 head->GetOutputHandler().SetTensorInfo(info);
52 head->GetOutputHandler().SetTensorInfo(info);
55 head->GetOutputHandler().SetTensorInfo(info);
58 head->GetOutputHandler().SetTensorInfo(info);
63 .SetTensorInfo(info);
68 .SetTensorInfo(info);
|
| D | AddBroadcastReshapeLayerTests.cpp | 32 input0->GetOutputSlot().SetTensorInfo(info0); in AddBroadcastReshapeLayerOptimizerTest()
33 input1->GetOutputSlot().SetTensorInfo(info1); in AddBroadcastReshapeLayerOptimizerTest()
34 add->GetOutputSlot().SetTensorInfo(outputInfo); in AddBroadcastReshapeLayerOptimizerTest()
126 input0->GetOutputSlot().SetTensorInfo(info0);
127 input1->GetOutputSlot().SetTensorInfo(info1);
128 sub->GetOutputSlot().SetTensorInfo(outputInfo);
171 input0->GetOutputSlot().SetTensorInfo(info0);
172 input1->GetOutputSlot().SetTensorInfo(info1);
173 div->GetOutputSlot().SetTensorInfo(outputInfo);
216 input0->GetOutputSlot().SetTensorInfo(info0);
[all …]
|
| D | FoldPadTests.cpp | 32 input->GetOutputSlot().SetTensorInfo(inputInfo);
40 padLayer->GetOutputSlot().SetTensorInfo(paddedInfo);
53 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
56 conv2dLayer->GetOutputSlot().SetTensorInfo(outputInfo);
114 input->GetOutputSlot().SetTensorInfo(inputInfo);
122 padLayer->GetOutputSlot().SetTensorInfo(paddedInfo);
134 weightsLayer->GetOutputSlot().SetTensorInfo(weightsInfo);
139 depthwiseConv2dLayer->GetOutputSlot().SetTensorInfo(outputInfo);
198 input->GetOutputSlot().SetTensorInfo(inputInfo);
206 padLayer->GetOutputSlot().SetTensorInfo(paddedInfo);
[all …]
|
| D | PermuteAndBatchToSpaceAsDepthToSpaceTests.cpp | 29 input->GetOutputSlot(0).SetTensorInfo(inputInfo); in CreateTestNetworkImpl()
34 permute->GetOutputSlot(0).SetTensorInfo(permuteInfo); in CreateTestNetworkImpl()
43 batchToSpace->GetOutputSlot(0).SetTensorInfo(batchToSpaceInfo); in CreateTestNetworkImpl()
60 input->GetOutputSlot(0).SetTensorInfo(inputInfo); in CreateTransposeTestNetworkImpl()
65 permute->GetOutputSlot(0).SetTensorInfo(permuteInfo); in CreateTransposeTestNetworkImpl()
74 batchToSpace->GetOutputSlot(0).SetTensorInfo(batchToSpaceInfo); in CreateTransposeTestNetworkImpl()
156 input->GetOutputSlot(0).SetTensorInfo(inputInfo); in CreateTestNetwork()
161 permute->GetOutputSlot(0).SetTensorInfo(permuteInfo); in CreateTestNetwork()
170 batchToSpace->GetOutputSlot(0).SetTensorInfo(batchToSpaceInfo); in CreateTestNetwork()
187 input->GetOutputSlot(0).SetTensorInfo(inputInfo); in CreateTransposeTestNetwork()
[all …]
|
| D | ConvertConstantsFloatToHalfTests.cpp | 33 input->GetOutputSlot().SetTensorInfo(info);
36 fc->GetOutputSlot().SetTensorInfo(info);
40 weightsLayer->GetOutputSlot().SetTensorInfo(weightsInfo);
96 input->GetOutputSlot().SetTensorInfo(inputInfo);
97 weights->GetOutputSlot().SetTensorInfo(weightsInfo);
98 biases->GetOutputSlot().SetTensorInfo(biasesInfo);
99 fcLayer->GetOutputSlot().SetTensorInfo(outputInfo);
|
| D | ConvertConstDequantisationLayersToConstLayersTest.cpp | 63 constantLayer->GetOutputSlot().SetTensorInfo(constTensorInfo);
66 dequantizeLayer->GetOutputSlot().SetTensorInfo(outputDequantizeInfo);
99 constantLayer->GetOutputSlot().SetTensorInfo(constTensorInfo);
102 dequantizeLayer->GetOutputSlot().SetTensorInfo(outputDequantizeInfo);
126 padLayer->GetOutputSlot().SetTensorInfo(paddedInfo);
169 constantLayer->GetOutputSlot().SetTensorInfo(constTensorInfo);
172 dequantizeLayer->GetOutputSlot().SetTensorInfo(outputDequantizeInfo);
|
| D | OptimizeConsecutiveReshapesTests.cpp | 25 input->GetOutputHandler().SetTensorInfo(info0);
40 reshape1->GetOutputHandler().SetTensorInfo(info1);
41 reshape2->GetOutputHandler().SetTensorInfo(info2);
71 reshapeToIn->GetOutputHandler().SetTensorInfo(info0);
|
| /external/armnn/src/backends/reference/test/ |
| D | RefOptimizedNetworkTests.cpp | 43 layer->GetOutputSlot(0).SetTensorInfo(desc);
48 normLayer->GetOutputSlot(0).SetTensorInfo(desc);
53 layer->GetOutputSlot(0).SetTensorInfo(desc);
60 layer->GetOutputSlot(0).SetTensorInfo(desc);
67 layer->GetOutputSlot(0).SetTensorInfo(desc);
112 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
113 …permute->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 4, 1, 4 }, armnn::DataType::Float32…
148 input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 4, 3, 2 }, armnn::DataType::Float32));
149 meanLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 2 }, armnn::DataType::Float32));
182 input->GetOutputSlot(0).SetTensorInfo(info);
[all …]
|
| /external/armnn/src/backends/backendsCommon/test/ |
| D | OptimizedNetworkTests.cpp | 34 input->GetOutputSlot(0).SetTensorInfo(info);
35 add->GetOutputSlot(0).SetTensorInfo(info);
85 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
86 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
122 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
123 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
177 layer->GetOutputSlot(0).SetTensorInfo(desc);
182 normLayer->GetOutputSlot(0).SetTensorInfo(desc);
187 layer->GetOutputSlot(0).SetTensorInfo(desc);
194 layer->GetOutputSlot(0).SetTensorInfo(desc);
[all …]
|
| D | EndToEndTestImpl.hpp | 52 input->GetOutputSlot(0).SetTensorInfo(commonTensorInfo); in ConstantUsageTest()
53 constant->GetOutputSlot(0).SetTensorInfo(commonTensorInfo); in ConstantUsageTest()
54 add->GetOutputSlot(0).SetTensorInfo(commonTensorInfo); in ConstantUsageTest()
209 …input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32, 0.0f, 0, true)… in ImportNonAlignedInputPointerTest()
210 pooling->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32)); in ImportNonAlignedInputPointerTest()
277 …input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32, 0.0f, 0, true)… in ExportNonAlignedOutputPointerTest()
278 pooling->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32)); in ExportNonAlignedOutputPointerTest()
352 …input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32, 0.0f, 0, true)… in ImportAlignedPointerTest()
353 pooling->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32)); in ImportAlignedPointerTest()
440 …input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 1, 4 }, DataType::Float32, 0.0f, 0, true)… in ImportOnlyWorkload()
[all …]
|
| D | LayerReleaseConstantDataTest.cpp | 95 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
96 biasLayer->GetOutputSlot(0).SetTensorInfo(biasInfo);
144 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
145 biasLayer->GetOutputSlot(0).SetTensorInfo(biasInfo);
191 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
192 biasLayer->GetOutputSlot(0).SetTensorInfo(biasInfo);
|
| /external/armnn/src/backends/neon/test/ |
| D | NeonFallbackTests.cpp | 53 input0->GetOutputSlot(0).SetTensorInfo(info);
54 input1->GetOutputSlot(0).SetTensorInfo(info);
55 input2->GetOutputSlot(0).SetTensorInfo(info);
56 add->GetOutputSlot(0).SetTensorInfo(info);
57 sub->GetOutputSlot(0).SetTensorInfo(info);
198 input0->GetOutputSlot(0).SetTensorInfo(info);
199 input1->GetOutputSlot(0).SetTensorInfo(info);
200 add->GetOutputSlot(0).SetTensorInfo(info);
201 pooling->GetOutputSlot(0).SetTensorInfo(poolingInfo);
333 input0->GetOutputSlot(0).SetTensorInfo(info);
[all …]
|
| D | NeonOptimizedNetworkTests.cpp | 26 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
63 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
64 …normalize->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float…
93 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
123 …input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({ 1, 1, 4, 4 }, armnn::DataType::Float32));
|
| /external/armnn/src/backends/cl/test/ |
| D | ClFallbackTests.cpp | 40 input0->GetOutputSlot(0).SetTensorInfo(info);
41 input1->GetOutputSlot(0).SetTensorInfo(info);
42 input2->GetOutputSlot(0).SetTensorInfo(info);
43 add->GetOutputSlot(0).SetTensorInfo(info);
44 sub->GetOutputSlot(0).SetTensorInfo(info);
188 input0->GetOutputSlot(0).SetTensorInfo(info);
189 input1->GetOutputSlot(0).SetTensorInfo(info);
190 input2->GetOutputSlot(0).SetTensorInfo(info);
191 add->GetOutputSlot(0).SetTensorInfo(info);
192 sub->GetOutputSlot(0).SetTensorInfo(info);
[all …]
|
| D | Fp16SupportTest.cpp | 39 inputLayer1->GetOutputSlot().SetTensorInfo(fp16TensorInfo);
40 inputLayer2->GetOutputSlot().SetTensorInfo(fp16TensorInfo);
41 additionLayer->GetOutputSlot().SetTensorInfo(fp16TensorInfo);
69 inputLayer1->GetOutputSlot(0).SetTensorInfo(fp16TensorInfo);
70 inputLayer2->GetOutputSlot(0).SetTensorInfo(fp16TensorInfo);
71 additionLayer->GetOutputSlot(0).SetTensorInfo(fp16TensorInfo);
|
| /external/armnn/src/backends/tosaCommon/test/ |
| D | OneToOneMappingTests.cpp | 48 input0->GetOutputSlot(0).SetTensorInfo(info);
49 input1->GetOutputSlot(0).SetTensorInfo(info);
50 add->GetOutputSlot(0).SetTensorInfo(info);
120 input0->GetOutputSlot(0).SetTensorInfo(inputInfo0);
121 input1->GetOutputSlot(0).SetTensorInfo(inputInfo1);
122 concat->GetOutputSlot(0).SetTensorInfo(outputInfo);
166 constant->GetOutputSlot(0).SetTensorInfo(info);
248 inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
249 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
250 biasesLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
[all …]
|
| /external/armnn/src/armnn/test/ |
| D | FlowControl.cpp | 32 constLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean));
35 input->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean));
38 switchLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean));
39 switchLayer->GetOutputSlot(1).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean));
42 mergeLayer->GetOutputSlot(0).SetTensorInfo(armnn::TensorInfo({1}, armnn::DataType::Boolean));
|
| D | OptimizerTests.cpp | 343 head->GetOutputHandler().SetTensorInfo(info);
346 ->GetOutputHandler().SetTensorInfo(info);
349 head->GetOutputHandler().SetTensorInfo(info);
352 head->GetOutputHandler().SetTensorInfo(info);
355 ->GetOutputHandler().SetTensorInfo(info);
442 input->GetOutputSlot().SetTensorInfo(inputInfo); in CreateConvolution2dGraph()
446 weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsLayer->m_LayerOutput->GetTensorInfo()); in CreateConvolution2dGraph()
449 layer->GetOutputSlot().SetTensorInfo(outputInfo); in CreateConvolution2dGraph()
502 input->GetOutputSlot().SetTensorInfo(inputInfo); in CreateDepthwiseConvolution2dGraph()
503 layer->GetOutputSlot().SetTensorInfo(outputInfo); in CreateDepthwiseConvolution2dGraph()
[all …]
|
| D | RuntimeTests.cpp | 88 inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);
91 inputLayer2->GetOutputSlot(0).SetTensorInfo(tensorInfo);
94 addLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
193 inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);
205 activationLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);
215 inputLayer2->GetOutputSlot(0).SetTensorInfo(tensorInfo);
218 activationLayer2->GetOutputSlot(0).SetTensorInfo(tensorInfo);
465 input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 4, 4 }, DataType::Float32));
466 normalize->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 4, 4 }, DataType::Float32));
499 input->GetOutputSlot(0).SetTensorInfo(TensorInfo({ 1, 1, 4, 4 }, DataType::Float32));
[all …]
|
| D | ShapeInferenceTests.cpp | 41 input->GetOutputSlot().SetTensorInfo(inputTensorInfo); in BuildGraph()
68 …layer->GetOutputSlot(i).SetTensorInfo({{numDimensions[i], dimensionSizeLists[i].begin(), maskPermu… in RunShapeInferenceTest()
83 …layer->GetOutputSlot(j).SetTensorInfo({TensorShape(Dimensionality::NotSpecified), DataType::Float3… in RunShapeInferenceTest()
131 inputLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
138 …activationLayer->GetOutputSlot(0).SetTensorInfo({TensorShape{Dimensionality::NotSpecified}, DataTy…
153 inputLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
158 …activationLayer->GetOutputSlot(0).SetTensorInfo({TensorShape{Dimensionality::NotSpecified}, DataTy…
165 inputLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
170 …activationLayer->GetOutputSlot(0).SetTensorInfo({TensorShape{Dimensionality::NotSpecified}, DataTy…
266 layer->GetOutputSlot(0).SetTensorInfo({{1, 1, 3, 3}, DataType::Float32});
[all …]
|
| /external/armnn/src/armnnDeserializer/ |
| D | Deserializer.cpp | 1048 inputLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo); in SetupInputLayers()
1210 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAbs()
1238 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseActivation()
1259 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseAdd()
1285 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseArgMinMax()
1316 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchMatMul()
1358 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseBatchToSpaceNd()
1395 layer->GetOutputSlot(0).SetTensorInfo(outputInfo); in ParseBatchNormalization()
1416 layer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo); in ParseCast()
1466 layer->GetOutputSlot(0).SetTensorInfo(weightsPermuted.GetInfo()); in ParseConstant()
[all …]
|
| /external/armnn/delegate/classic/src/ |
| D | Convolution.hpp | 192 weightsLayer->GetOutputSlot(0).SetTensorInfo(filterTensorInfo); in VisitConv2dOperator()
204 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensorInfo); in VisitConv2dOperator()
217 inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo); in VisitConv2dOperator()
223 outputSlot.SetTensorInfo(outputTensorInfo); in VisitConv2dOperator()
380 weightsLayer->GetOutputSlot(0).SetTensorInfo(filterTensorInfo); in VisitConv3dOperator()
395 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensorInfo); in VisitConv3dOperator()
408 inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo); in VisitConv3dOperator()
412 outputSlot.SetTensorInfo(outputTensorInfo); in VisitConv3dOperator()
600 weightsLayer->GetOutputSlot(0).SetTensorInfo(filterTensorInfo); in VisitDepthwiseConv2dOperator()
612 biasLayer->GetOutputSlot(0).SetTensorInfo(biasTensorInfo); in VisitDepthwiseConv2dOperator()
[all …]
|