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