aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/QuantizerVisitor.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/QuantizerVisitor.cpp')
-rw-r--r--src/armnn/QuantizerVisitor.cpp9
1 files changed, 9 insertions, 0 deletions
diff --git a/src/armnn/QuantizerVisitor.cpp b/src/armnn/QuantizerVisitor.cpp
index 9c1ac17d70..a30a29db75 100644
--- a/src/armnn/QuantizerVisitor.cpp
+++ b/src/armnn/QuantizerVisitor.cpp
@@ -284,6 +284,15 @@ void QuantizerVisitor::VisitElementwiseUnaryLayer(const IConnectableLayer* layer
SetQuantizedInputConnections(layer, newLayer);
}
+void QuantizerVisitor::VisitFillLayer(const IConnectableLayer* layer,
+ const FillDescriptor& desc,
+ const char* name)
+{
+ IConnectableLayer* newLayer = m_QuantizedNetwork->AddFillLayer(desc, name);
+ RecordLayer(layer, newLayer);
+ SetQuantizedInputConnections(layer, newLayer);
+}
+
void QuantizerVisitor::VisitFullyConnectedLayer(const IConnectableLayer *layer,
const FullyConnectedDescriptor& desc,
const ConstTensor& weights,