diff options
Diffstat (limited to 'src/backends/backendsCommon/WorkloadFactory.cpp')
-rw-r--r-- | src/backends/backendsCommon/WorkloadFactory.cpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/WorkloadFactory.cpp b/src/backends/backendsCommon/WorkloadFactory.cpp index f067056ce1..6ff237bc12 100644 --- a/src/backends/backendsCommon/WorkloadFactory.cpp +++ b/src/backends/backendsCommon/WorkloadFactory.cpp @@ -525,6 +525,36 @@ bool IWorkloadFactory::IsLayerConfigurationSupported(const BackendId& backendId, reason); break; } + case LayerType::Fused: + { + auto cLayer = PolymorphicDowncast<const FusedLayer*>(&layer); + + // Get vector of all outputs. + auto getOutTensorInfo = [&dataType](const OutputSlot& slot) + { + return OverrideDataType(slot.GetTensorInfo(), dataType); + }; + auto beginOutputs = MakeTransformIterator(layer.GetOutputSlots().begin(), getOutTensorInfo); + auto endOutputs = MakeTransformIterator(layer.GetOutputSlots().end(), getOutTensorInfo); + std::vector<TensorInfo> outputs(beginOutputs, endOutputs); + const std::vector<std::reference_wrapper<TensorInfo>> outputPtrs(outputs.begin(), outputs.end()); + + // Get vector of all inputs. + auto getInputTensorInfo = [&dataType](const InputSlot& slot) + { + return OverrideDataType(slot.GetTensorInfo(), dataType); + }; + auto beginInputs = MakeTransformIterator(layer.GetInputSlots().begin(), getInputTensorInfo); + auto endInputs = MakeTransformIterator(layer.GetInputSlots().end(), getInputTensorInfo); + std::vector<TensorInfo> inputs(beginInputs, endInputs); + const std::vector<std::reference_wrapper<TensorInfo>> inputPtrs(inputs.begin(), inputs.end()); + + result = layerSupportObject.IsFusedSupported(inputPtrs, + outputPtrs, + cLayer->GetParameters(), + reason); + break; + } case LayerType::Gather: { const TensorInfo& input0 = layer.GetInputSlot(0).GetTensorInfo(); |