SerializerTests.cpp - OpenGrok cross reference for /external/armnn/src/armnnSerializer/test/SerializerTests.cpp

Lines Matching refs:SetTensorInfo
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);
208     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
209     batchToSpaceNdLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
238         inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
239         castLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
265     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
266     ChannelShuffleLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
297     inputLayer0->GetOutputSlot(0).SetTensorInfo(inputInfo);
298     inputLayer1->GetOutputSlot(0).SetTensorInfo(inputInfo);
299     comparisonLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
371     input->GetOutputSlot(0).SetTensorInfo(info);
372     constant->GetOutputSlot(0).SetTensorInfo(info);
373     add->GetOutputSlot(0).SetTensorInfo(info);
454     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
455     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
456     biasLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
457     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
509     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
510     weightsLayer->GetOutputSlot(0).SetTensorInfo(kernelInfo);
511     biasLayer->GetOutputSlot(0).SetTensorInfo(biasInfo);
512     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
562     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
563     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
564     biasesLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
565     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
620     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
621     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
622     biasesLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
623     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
652     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
653     depthToSpaceLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
698     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
699     depthwiseConvLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
703     weightsLayer->GetOutputSlot(0).SetTensorInfo(weights.GetInfo());
707     biasLayer->GetOutputSlot(0).SetTensorInfo(biases.GetInfo());
760     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
761     depthwiseConvLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
765     weightsLayer->GetOutputSlot(0).SetTensorInfo(weights.GetInfo());
769     biasLayer->GetOutputSlot(0).SetTensorInfo(biases.GetInfo());
820     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
821     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
822     biasesLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
823     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
849     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
850     dequantizeLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
907         inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfos[i]);
914         detectionLayer->GetOutputSlot(i).SetTensorInfo(outputInfos[i]);
943     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
944     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
945     divisionLayer->GetOutputSlot(0).SetTensorInfo(info);
969     inputLayer1->GetOutputSlot(0).SetTensorInfo(inputTensorInfo1);
971     inputLayer2->GetOutputSlot(0).SetTensorInfo(inputTensorInfo2);
973     equalLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
999     inputLayer0->GetOutputSlot(0).SetTensorInfo(tensorInfo);  in SerializeElementwiseBinaryTest()
1000     inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);  in SerializeElementwiseBinaryTest()
1001     elementwiseBinaryLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);  in SerializeElementwiseBinaryTest()
1043     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);  in SerializeElementwiseUnaryTest()
1044     elementwiseUnaryLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);  in SerializeElementwiseUnaryTest()
1084     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
1085     fillLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1108     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
1109     floorLayer->GetOutputSlot(0).SetTensorInfo(info);
1185     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
1186     weightsInputLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
1187     biasInputLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
1188     fullyConnectedLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1228     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
1229     weightsInputLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
1230     biasInputLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
1231     fullyConnectedLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1273     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
1274     weightsLayer->GetOutputSlot(0).SetTensorInfo(weightsInfo);
1275     biasesLayer->GetOutputSlot(0).SetTensorInfo(biasesInfo);
1276     fullyConnectedLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1344     inputLayer->GetOutputSlot(0).SetTensorInfo(paramsInfo);
1345     constantLayer->GetOutputSlot(0).SetTensorInfo(indicesInfo);
1346     gatherLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1409     inputLayer->GetOutputSlot(0).SetTensorInfo(paramsInfo);
1410     constantLayer->GetOutputSlot(0).SetTensorInfo(indicesInfo);
1411     gatherNdLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1440     inputLayer0->GetOutputSlot(0).SetTensorInfo(inputInfo);
1441     inputLayer1->GetOutputSlot(0).SetTensorInfo(inputInfo);
1442     equalLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1472     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
1473     instanceNormLayer->GetOutputSlot(0).SetTensorInfo(info);
1500     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
1501     l2NormLayer->GetOutputSlot(0).SetTensorInfo(info);
1589     inputLayer0->GetOutputSlot(0).SetTensorInfo(inputInfo);
1590     inputLayer1->GetOutputSlot(0).SetTensorInfo(inputInfo);
1591     logicalBinaryLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1618     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
1619     logSoftmaxLayer->GetOutputSlot(0).SetTensorInfo(info);
1645     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
1646     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
1647     maximumLayer->GetOutputSlot(0).SetTensorInfo(info);
1674     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
1675     meanLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1699     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
1700     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
1701     mergeLayer->GetOutputSlot(0).SetTensorInfo(info);
1841     inputLayerOne->GetOutputSlot(0).SetTensorInfo(inputInfo);
1842     inputLayerTwo->GetOutputSlot(0).SetTensorInfo(inputInfo);
1843     concatLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
1872     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
1873     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
1874     minimumLayer->GetOutputSlot(0).SetTensorInfo(info);
1900     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
1901     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
1902     multiplicationLayer->GetOutputSlot(0).SetTensorInfo(info);
1929     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
1930     alphaLayer->GetOutputSlot(0).SetTensorInfo(alphaTensorInfo);
1931     preluLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
1960     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
1961     normalizationLayer->GetOutputSlot(0).SetTensorInfo(info);
1986     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
1987     padLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
2016     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2017     padLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
2099     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2100     permuteLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
2138     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2139     pooling2dLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2181     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2182     pooling3dLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2205     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
2206     quantizeLayer->GetOutputSlot(0).SetTensorInfo(info);
2229     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2230     rankLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2257     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2258     reduceSumLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2283     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2284     reshapeLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2315     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2316     resizeLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2389     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2390     resizeLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2469     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2470     shapeLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2498     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2499     sliceLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2524     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
2525     softmaxLayer->GetOutputSlot(0).SetTensorInfo(info);
2553     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2554     spaceToBatchNdLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2583     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2584     spaceToDepthLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2635     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2636     splitterLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2637     splitterLayer->GetOutputSlot(1).SetTensorInfo(outputInfo);
2638     splitterLayer->GetOutputSlot(2).SetTensorInfo(outputInfo);
2667     inputLayer1->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2668     inputLayer2->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2669     stackLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
2694     inputLayer0->GetOutputSlot(0).SetTensorInfo(tensorInfo);
2697     inputLayer1->GetOutputSlot(0).SetTensorInfo(tensorInfo);
2700     standInLayer->GetOutputSlot(0).SetTensorInfo(tensorInfo);
2703     standInLayer->GetOutputSlot(1).SetTensorInfo(tensorInfo);
2732     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2733     stridedSliceLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2760     inputLayer0->GetOutputSlot(0).SetTensorInfo(info);
2761     inputLayer1->GetOutputSlot(0).SetTensorInfo(info);
2762     subtractionLayer->GetOutputSlot(0).SetTensorInfo(info);
2824     inputLayer->GetOutputSlot(0).SetTensorInfo(info);
2825     constantLayer->GetOutputSlot(0).SetTensorInfo(info);
2826     switchLayer->GetOutputSlot(0).SetTensorInfo(info);
2827     switchLayer->GetOutputSlot(1).SetTensorInfo(info);
2852     inputLayer->GetOutputSlot(0).SetTensorInfo(inputTensorInfo);
2853     transposeLayer->GetOutputSlot(0).SetTensorInfo(outputTensorInfo);
2900     inputLayer->GetOutputSlot(0).SetTensorInfo(inputInfo);
2901     convLayer->GetOutputSlot(0).SetTensorInfo(outputInfo);
2972     input->GetOutputSlot(0).SetTensorInfo(info);
2973     constant->GetOutputSlot(0).SetTensorInfo(info);
2974     add->GetOutputSlot(0).SetTensorInfo(info);