Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 1 | // |
| 2 | // Copyright © 2020 Arm Ltd and Contributors. All rights reserved. |
| 3 | // SPDX-License-Identifier: MIT |
| 4 | // |
| 5 | #pragma once |
| 6 | |
| 7 | #include <armnn/backends/IBackendContext.hpp> |
| 8 | |
Sadik Armagan | dea8fb6 | 2020-11-26 10:38:11 +0000 | [diff] [blame] | 9 | #include<string> |
| 10 | |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 11 | namespace armnn |
| 12 | { |
| 13 | |
Mike Kelly | f9f33a0 | 2020-10-14 11:48:21 +0100 | [diff] [blame] | 14 | /// The ClBackendModelContext is used to pass in CL specific backend ModelOptions. The supported backend ModelOptions |
| 15 | /// are: |
| 16 | /// - "FastMathEnabled"\n |
| 17 | /// Using the fast_math flag can lead to performance improvements in fp32 and fp16 layers but may result in\n |
| 18 | /// results with reduced or different precision. The fast_math flag will not have any effect on int8 performance. |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 19 | class ClBackendModelContext : public IBackendModelContext |
| 20 | { |
| 21 | public: |
| 22 | ClBackendModelContext(const ModelOptions& modelOptions); |
| 23 | |
Sadik Armagan | dea8fb6 | 2020-11-26 10:38:11 +0000 | [diff] [blame] | 24 | std::string GetCachedNetworkFilePath() const; |
| 25 | |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 26 | bool IsFastMathEnabled() const; |
| 27 | |
Sadik Armagan | dea8fb6 | 2020-11-26 10:38:11 +0000 | [diff] [blame] | 28 | bool SaveCachedNetwork() const; |
| 29 | |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 30 | private: |
Sadik Armagan | dea8fb6 | 2020-11-26 10:38:11 +0000 | [diff] [blame] | 31 | std::string m_CachedNetworkFilePath; |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 32 | bool m_IsFastMathEnabled; |
Sadik Armagan | dea8fb6 | 2020-11-26 10:38:11 +0000 | [diff] [blame] | 33 | bool m_SaveCachedNetwork; |
| 34 | |
Sadik Armagan | 045f6be | 2020-09-10 13:37:32 +0100 | [diff] [blame] | 35 | }; |
| 36 | |
| 37 | } // namespace armnn |