diff options
Diffstat (limited to 'src/backends/WorkloadFactory.cpp')
| -rw-r--r-- | src/backends/WorkloadFactory.cpp | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/backends/WorkloadFactory.cpp b/src/backends/WorkloadFactory.cpp index d23ac83230..aaccabd059 100644 --- a/src/backends/WorkloadFactory.cpp +++ b/src/backends/WorkloadFactory.cpp @@ -274,10 +274,18 @@ bool IWorkloadFactory::IsLayerSupported(Compute compute, const Layer& layer, boo } case LayerType::L2Normalization: { + auto cLayer = boost::polymorphic_downcast<const L2NormalizationLayer*>(&layer); + const L2NormalizationDescriptor& descriptor = cLayer->GetParameters(); + const TensorInfo& input = layer.GetInputSlot(0).GetConnection()->GetTensorInfo(); const TensorInfo& output = layer.GetOutputSlot(0).GetTensorInfo(); - result = IsL2NormalizationSupported(compute, OverrideDataType(input, dataType), - OverrideDataType(output, dataType), reason, reasonCapacity); + + result = IsL2NormalizationSupported(compute, + OverrideDataType(input, dataType), + OverrideDataType(output, dataType), + descriptor, + reason, + reasonCapacity); break; } case LayerType::Lstm: |