diff options
Diffstat (limited to 'src/armnn/backends/WorkloadFactory.hpp')
| -rw-r--r-- | src/armnn/backends/WorkloadFactory.hpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/src/armnn/backends/WorkloadFactory.hpp b/src/armnn/backends/WorkloadFactory.hpp index d3f5bfb40f..5791c1b46f 100644 --- a/src/armnn/backends/WorkloadFactory.hpp +++ b/src/armnn/backends/WorkloadFactory.hpp @@ -22,8 +22,11 @@ public: virtual Compute GetCompute() const = 0; + /// Informs the memory manager that the network is finalized and ready for execution. + virtual void Finalize() { } + static bool IsLayerSupported(Compute compute, const Layer& layer, DataType dataType, - std::string& outReasonIfUnsupported); + std::string& outReasonIfUnsupported); static bool IsLayerSupported(const Layer& layer, DataType dataType, std::string& outReasonIfUnsupported); virtual bool SupportsSubTensors() const = 0; @@ -102,4 +105,4 @@ public: const WorkloadInfo& info) const = 0; }; -} //namespace armnn
\ No newline at end of file +} //namespace armnn |