• Home
  • Raw
  • Download

Lines Matching full:layer

6 #include <Layer.hpp>
26 using LayerList = std::list<Layer*>;
78 const Layer& layer = *(PolymorphicDowncast<const Layer*>(&connectableLayer)); in IsLayerConfigurationSupported() local
96 switch(layer.GetType()) in IsLayerConfigurationSupported()
100 auto cLayer = PolymorphicDowncast<const ActivationLayer*>(&layer); in IsLayerConfigurationSupported()
101 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
102 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
113 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
114 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
115 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
126 auto cLayer = PolymorphicDowncast<const ArgMinMaxLayer*>(&layer); in IsLayerConfigurationSupported()
129 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
130 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
140 auto cLayer = PolymorphicDowncast<const BatchMatMulLayer*>(&layer); in IsLayerConfigurationSupported()
143 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
144 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
145 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
156 auto cLayer = PolymorphicDowncast<const BatchNormalizationLayer*>(&layer); in IsLayerConfigurationSupported()
157 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
158 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
176 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
177 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
178 auto cLayer = PolymorphicDowncast<const BatchToSpaceNdLayer*>(&layer); in IsLayerConfigurationSupported()
188 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
189 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
198 auto cLayer = PolymorphicDowncast<const ChannelShuffleLayer*>(&layer); in IsLayerConfigurationSupported()
200 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
201 const TensorInfo& output = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
213 auto cLayer = PolymorphicDowncast<const ComparisonLayer*>(&layer); in IsLayerConfigurationSupported()
215 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
216 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
217 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
228 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
234 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
235 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
241 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
242 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
248 auto cLayer = PolymorphicDowncast<const Convolution2dLayer*>(&layer); in IsLayerConfigurationSupported()
250 … const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
252 … const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
253 ARMNN_ASSERT_MSG(layer.GetInputSlot(1).GetConnection(), in IsLayerConfigurationSupported()
254 … "Convolution2dLayer: Weights should be connected as a Constant Layer."); in IsLayerConfigurationSupported()
255 …const TensorInfo weights = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
264 ARMNN_ASSERT_MSG(layer.GetInputSlot(2).GetConnection(), in IsLayerConfigurationSupported()
265 … "Convolution2dLayer: Bias should be connected as a Constant Layer."); in IsLayerConfigurationSupported()
266 biases = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
281 auto cLayer = PolymorphicDowncast<const Convolution3dLayer*>(&layer); in IsLayerConfigurationSupported()
283 … const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
285 … const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
287 ARMNN_ASSERT_MSG(layer.GetInputSlot(1).GetConnection(), in IsLayerConfigurationSupported()
288 … "Convolution3dLayer: Weights should be connected as a Constant Layer."); in IsLayerConfigurationSupported()
289 …const TensorInfo weights = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
298 biases = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
313 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
314 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
323 auto cLayer = PolymorphicDowncast<const DepthToSpaceLayer*>(&layer); in IsLayerConfigurationSupported()
325 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
326 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
336 auto cLayer = PolymorphicDowncast<const DepthwiseConvolution2dLayer*>(&layer); in IsLayerConfigurationSupported()
337 …const TensorInfo& input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(… in IsLayerConfigurationSupported()
339 … const TensorInfo& output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
340 …const TensorInfo& weights = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(… in IsLayerConfigurationSupported()
365 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
366 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
375 auto cLayer = PolymorphicDowncast<const DetectionPostProcessLayer*>(&layer); in IsLayerConfigurationSupported()
376 const TensorInfo& boxEncodings = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
377 const TensorInfo& scores = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
380 const TensorInfo& detectionBoxes = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
381 const TensorInfo& detectionClasses = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported()
382 const TensorInfo& detectionScores = layer.GetOutputSlot(2).GetTensorInfo(); in IsLayerConfigurationSupported()
383 const TensorInfo& numDetections = layer.GetOutputSlot(3).GetTensorInfo(); in IsLayerConfigurationSupported()
399 auto cLayer = PolymorphicDowncast<const ElementwiseBinaryLayer*>(&layer); in IsLayerConfigurationSupported()
401 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
402 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
403 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
417 auto cLayer = PolymorphicDowncast<const ElementwiseUnaryLayer*>(&layer); in IsLayerConfigurationSupported()
419 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
420 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
430 auto cLayer = PolymorphicDowncast<const FillLayer*>(&layer); in IsLayerConfigurationSupported()
431 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
432 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
444 auto cLayer = PolymorphicDowncast<const FakeQuantizationLayer*>(&layer); in IsLayerConfigurationSupported()
445 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
453 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
454 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
462 auto cLayer = PolymorphicDowncast<const FullyConnectedLayer*>(&layer); in IsLayerConfigurationSupported()
463 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
464 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
470 … weightsInfo = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
482 … biasInfo = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
530 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
531 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
532 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
533 auto cLayer = PolymorphicDowncast<const GatherLayer*>(&layer); in IsLayerConfigurationSupported()
544 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
545 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
546 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
555 const TensorInfo& input = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
561 auto cLayer = PolymorphicDowncast<const InstanceNormalizationLayer*>(&layer); in IsLayerConfigurationSupported()
564 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
565 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
576 auto cLayer = PolymorphicDowncast<const L2NormalizationLayer*>(&layer); in IsLayerConfigurationSupported()
579 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
580 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
591 auto cLayer = PolymorphicDowncast<const LogicalBinaryLayer*>(&layer); in IsLayerConfigurationSupported()
593 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
594 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
595 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
606 auto cLayer = PolymorphicDowncast<const LogSoftmaxLayer*>(&layer); in IsLayerConfigurationSupported()
608 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
609 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
619 auto cLayer = PolymorphicDowncast<const LstmLayer*>(&layer); in IsLayerConfigurationSupported()
623 … const TensorInfo& input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
625 …const TensorInfo& outputStateIn = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTenso… in IsLayerConfigurationSupported()
627 …const TensorInfo& cellStateIn = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorI… in IsLayerConfigurationSupported()
630 …const TensorInfo& scratchBuffer = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataTyp… in IsLayerConfigurationSupported()
631 …const TensorInfo& outputStateOut = OverrideDataType(layer.GetOutputSlot(1).GetTensorInfo(), dataTy… in IsLayerConfigurationSupported()
632 …const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(2).GetTensorInfo(), dataType… in IsLayerConfigurationSupported()
633 … const TensorInfo& output = OverrideDataType(layer.GetOutputSlot(3).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
764 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
765 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
766 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
777 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
778 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
787 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
788 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
797 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
798 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
799 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
809 auto cLayer = PolymorphicDowncast<const ConcatLayer*>(&layer); in IsLayerConfigurationSupported()
817 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
818 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
830 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
840 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
841 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
842 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
853 auto cLayer = PolymorphicDowncast<const NormalizationLayer*>(&layer); in IsLayerConfigurationSupported()
854 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
855 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
864 const TensorInfo& output = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
870 auto cLayer = PolymorphicDowncast<const PermuteLayer*>(&layer); in IsLayerConfigurationSupported()
871 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
872 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
881 auto cLayer = PolymorphicDowncast<const PadLayer*>(&layer); in IsLayerConfigurationSupported()
882 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
883 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
893 auto cLayer = PolymorphicDowncast<const Pooling2dLayer*>(&layer); in IsLayerConfigurationSupported()
894 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
895 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
904 auto cLayer = PolymorphicDowncast<const Pooling3dLayer*>(&layer); in IsLayerConfigurationSupported()
905 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
906 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
915 auto cLayer = PolymorphicDowncast<const PreCompiledLayer*>(&layer); in IsLayerConfigurationSupported()
916 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
924 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
925 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
931 auto cLayer = PolymorphicDowncast<const QLstmLayer*>(&layer); in IsLayerConfigurationSupported()
935 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
936 … const TensorInfo& previousOutputIn = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
937 … const TensorInfo& previousCellStateIn = layer.GetInputSlot(2).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
940 const TensorInfo& outputStateOut = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
941 const TensorInfo& cellStateOut = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported()
942 const TensorInfo& output = layer.GetOutputSlot(2).GetTensorInfo(); in IsLayerConfigurationSupported()
1027 auto cLayer = PolymorphicDowncast<const QuantizedLstmLayer*>(&layer); in IsLayerConfigurationSupported()
1030 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1031 … const TensorInfo& previousCellStateIn = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1032 … const TensorInfo& previousOutputIn = layer.GetInputSlot(2).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1035 const TensorInfo& cellStateOut = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1036 const TensorInfo& output = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported()
1080 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1081 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1082 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1093 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1094 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1102 auto cLayer = PolymorphicDowncast<const ReshapeLayer*>(&layer); in IsLayerConfigurationSupported()
1103 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1104 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1113 auto cLayer = PolymorphicDowncast<const ResizeLayer*>(&layer); in IsLayerConfigurationSupported()
1114 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1115 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1124 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1125 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1134 auto cLayer = PolymorphicDowncast<const SliceLayer*>(&layer); in IsLayerConfigurationSupported()
1136 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1137 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1147 auto cLayer = PolymorphicDowncast<const SoftmaxLayer*>(&layer); in IsLayerConfigurationSupported()
1148 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1149 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1158 auto cLayer = PolymorphicDowncast<const SpaceToBatchNdLayer*>(&layer); in IsLayerConfigurationSupported()
1159 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1160 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1169 auto cLayer = PolymorphicDowncast<const SpaceToDepthLayer*>(&layer); in IsLayerConfigurationSupported()
1171 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1172 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1182 auto cLayer = PolymorphicDowncast<const SplitterLayer*>(&layer); in IsLayerConfigurationSupported()
1183 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1190 auto beginI = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
1191 auto endI = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1204 auto cLayer = PolymorphicDowncast<const StackLayer*>(&layer); in IsLayerConfigurationSupported()
1211 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfo); in IsLayerConfigurationSupported()
1212 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfo); in IsLayerConfigurationSupported()
1223 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1231 auto cLayer = PolymorphicDowncast<const StandInLayer*>(&layer); in IsLayerConfigurationSupported()
1242 auto beginI = MakeTransformIterator(layer.GetInputSlots().begin(), getTensorInfoIn); in IsLayerConfigurationSupported()
1243 auto endI = MakeTransformIterator(layer.GetInputSlots().end(), getTensorInfoIn); in IsLayerConfigurationSupported()
1246 auto beginO = MakeTransformIterator(layer.GetOutputSlots().begin(), getTensorInfoOut); in IsLayerConfigurationSupported()
1247 auto endO = MakeTransformIterator(layer.GetOutputSlots().end(), getTensorInfoOut); in IsLayerConfigurationSupported()
1272 auto cLayer = PolymorphicDowncast<const StridedSliceLayer*>(&layer); in IsLayerConfigurationSupported()
1273 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1274 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1284 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1285 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1286 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1297 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1298 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1299 const TensorInfo& output0 = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1300 const TensorInfo& output1 = layer.GetOutputSlot(1).GetTensorInfo(); in IsLayerConfigurationSupported()
1310 auto cLayer = PolymorphicDowncast<const MeanLayer*>(&layer); in IsLayerConfigurationSupported()
1311 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1312 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1323 const TensorInfo& input0 = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1324 const TensorInfo& input1 = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1325 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1335 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1336 const TensorInfo& alpha = layer.GetInputSlot(1).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1337 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1346 auto cLayer = PolymorphicDowncast<const TransposeLayer*>(&layer); in IsLayerConfigurationSupported()
1347 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1348 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1357 auto cLayer = PolymorphicDowncast<const TransposeConvolution2dLayer*>(&layer); in IsLayerConfigurationSupported()
1359 … const TensorInfo input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
1361 … const TensorInfo output = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
1387 auto cLayer = PolymorphicDowncast<const ReduceLayer*>(&layer); in IsLayerConfigurationSupported()
1388 const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); in IsLayerConfigurationSupported()
1389 const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); in IsLayerConfigurationSupported()
1399 auto cLayer = PolymorphicDowncast<const UnidirectionalSequenceLstmLayer*>(&layer); in IsLayerConfigurationSupported()
1403 … const TensorInfo& input = OverrideDataType(layer.GetInputSlot(0).GetConnection()->GetTensorInfo(), in IsLayerConfigurationSupported()
1405 …const TensorInfo& outputStateIn = OverrideDataType(layer.GetInputSlot(1).GetConnection()->GetTenso… in IsLayerConfigurationSupported()
1407 …const TensorInfo& cellStateIn = OverrideDataType(layer.GetInputSlot(2).GetConnection()->GetTensorI… in IsLayerConfigurationSupported()
1410 …const TensorInfo& outputStateOut = OverrideDataType(layer.GetOutputSlot(0).GetTensorInfo(), dataTy… in IsLayerConfigurationSupported()
1411 …const TensorInfo& cellStateOut = OverrideDataType(layer.GetOutputSlot(1).GetTensorInfo(), dataType… in IsLayerConfigurationSupported()
1412 … const TensorInfo& output = OverrideDataType(layer.GetOutputSlot(2).GetTensorInfo(), dataType); in IsLayerConfigurationSupported()
1539 ARMNN_ASSERT_MSG(false, "WorkloadFactory did not recognise type of layer."); in IsLayerConfigurationSupported()
1540 reason.value() = "Unrecognised layer type"; in IsLayerConfigurationSupported()
1560 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer); in IsLayerSupported() local
1561 …return IsLayerConfigurationSupported(layer->GetBackendId(), connectableLayer, dataType, outReasonI… in IsLayerSupported()
1570 auto layer = PolymorphicDowncast<const Layer*>(&connectableLayer); in IsLayerSupported() local
1571 return IsLayerConfigurationSupported(layer->GetBackendId(), in IsLayerSupported()