ArmNN
 24.08
QuantizeLayer.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
6 #pragma once
7 
8 #include <Layer.hpp>
9 
10 namespace armnn {
11 
12 //Forward
13 class IWorkload;
14 class IWorkloadFactory;
15 
16 class QuantizeLayer : public Layer
17 {
18 public:
19  virtual std::unique_ptr<IWorkload> CreateWorkload(const IWorkloadFactory& factory) const override;
20 
21  Layer* Clone(Graph& graph) const override;
22 
23  void ValidateTensorShapesFromInputs() override;
24 
25  void ExecuteStrategy(IStrategy& strategy) const override;
26 
27 
28 protected:
29  QuantizeLayer(const char* name);
30  ~QuantizeLayer() = default;
31 
32 };
33 
34 } //namespace armnn
armnn::QuantizeLayer::Clone
Layer * Clone(Graph &graph) const override
Creates a dynamically-allocated copy of this layer.
Definition: QuantizeLayer.cpp:27
armnn::IStrategy
Definition: IStrategy.hpp:16
armnn::QuantizeLayer::QuantizeLayer
QuantizeLayer(const char *name)
Definition: QuantizeLayer.cpp:13
armnn::Layer
Definition: Layer.hpp:230
armnn::IWorkloadFactory
Definition: WorkloadFactory.hpp:22
armnn::QuantizeLayer
Definition: QuantizeLayer.hpp:16
armnn::QuantizeLayer::CreateWorkload
virtual std::unique_ptr< IWorkload > CreateWorkload(const IWorkloadFactory &factory) const override
Definition: QuantizeLayer.cpp:17
armnn::QuantizeLayer::~QuantizeLayer
~QuantizeLayer()=default
armnn::QuantizeLayer::ValidateTensorShapesFromInputs
void ValidateTensorShapesFromInputs() override
Definition: QuantizeLayer.cpp:33
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
Layer.hpp
armnn::Graph
Definition: Graph.hpp:30
armnn::QuantizeLayer::ExecuteStrategy
void ExecuteStrategy(IStrategy &strategy) const override
Apply a visitor to this layer.
Definition: QuantizeLayer.cpp:46