IVGCVSW-4009 StandInLayer frontend API

Change-Id: I058c57b554769799c6775813215070ef47790e3d
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>

1 files changed, 41 insertions, 0 deletions

diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp
index 30dfa023f9..34e4cbe579 100644
--- a/src/backends/backendsCommon/WorkloadFactory.cpp
+++ b/src/backends/backendsCommon/WorkloadFactory.cpp
@@ -902,6 +902,47 @@ bool IWorkloadFactory::IsLayerSupported(const BackendId& backendId,
 
             break;
         }
+        case LayerType::StandIn:
+        {
+            auto cLayer = boost::polymorphic_downcast<const StandInLayer*>(&layer);
+
+            // Get vector of all inputs.
+            auto getTensorInfoIn = [&dataType](const InputSlot& slot)
+                {
+                    return OverrideDataType(slot.GetConnectedOutputSlot()->GetTensorInfo(), dataType);
+                };
+            auto getTensorInfoOut = [&dataType](const OutputSlot& slot)
+                {
+                    return OverrideDataType(slot.GetTensorInfo(), dataType);
+                };
+            auto beginI = boost::make_transform_iterator(layer.GetInputSlots().begin(), getTensorInfoIn);
+            auto endI = boost::make_transform_iterator(layer.GetInputSlots().end(), getTensorInfoIn);
+            std::vector<TensorInfo> inputs(beginI, endI);
+
+            auto beginO = boost::make_transform_iterator(layer.GetOutputSlots().begin(), getTensorInfoOut);
+            auto endO = boost::make_transform_iterator(layer.GetOutputSlots().end(), getTensorInfoOut);
+            std::vector<TensorInfo> outputs(beginO, endO);
+
+
+            auto getTensorInfoPtr = [](const TensorInfo& info)
+                {
+                    return &info;
+                };
+            auto beginPtrI = boost::make_transform_iterator(inputs.begin(), getTensorInfoPtr);
+            auto endPtrI = boost::make_transform_iterator(inputs.end(), getTensorInfoPtr);
+            std::vector<const TensorInfo*> inputPtrs(beginPtrI, endPtrI);
+
+            auto beginPtrO = boost::make_transform_iterator(outputs.begin(), getTensorInfoPtr);
+            auto endPtrO = boost::make_transform_iterator(outputs.end(), getTensorInfoPtr);
+            std::vector<const TensorInfo*> outputPtrs(beginPtrO, endPtrO);
+
+
+            result = layerSupportObject->IsStandInSupported(inputPtrs,
+                                                            outputPtrs,
+                                                            cLayer->GetParameters(),
+                                                            reason);
+            break;
+        }
         case LayerType::StridedSlice:
         {
             auto cLayer = boost::polymorphic_downcast<const StridedSliceLayer*>(&layer);