ArmNN
 21.02
OptimizerOptions Struct Reference

#include <INetwork.hpp>

Public Member Functions

 OptimizerOptions ()
 
 OptimizerOptions (bool reduceFp32ToFp16, bool debug, bool reduceFp32ToBf16, bool importEnabled, ModelOptions modelOptions={})
 
 OptimizerOptions (bool reduceFp32ToFp16, bool debug, bool reduceFp32ToBf16=false, ShapeInferenceMethod shapeInferenceMethod=armnn::ShapeInferenceMethod::ValidateOnly, bool importEnabled=false, ModelOptions modelOptions={})
 

Public Attributes

bool m_ReduceFp32ToFp16
 
bool m_Debug
 
bool m_ReduceFp32ToBf16
 
ShapeInferenceMethod m_shapeInferenceMethod
 
bool m_ImportEnabled
 
ModelOptions m_ModelOptions
 

Detailed Description

Definition at line 110 of file INetwork.hpp.

Constructor & Destructor Documentation

◆ OptimizerOptions() [1/3]

OptimizerOptions ( )
inline

Definition at line 112 of file INetwork.hpp.

113  : m_ReduceFp32ToFp16(false)
114  , m_Debug(false)
115  , m_ReduceFp32ToBf16(false)
117  , m_ImportEnabled(false)
118  , m_ModelOptions()
119  {}
ModelOptions m_ModelOptions
Definition: INetwork.hpp:168
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:162
Validate all output shapes.

◆ OptimizerOptions() [2/3]

OptimizerOptions ( bool  reduceFp32ToFp16,
bool  debug,
bool  reduceFp32ToBf16,
bool  importEnabled,
ModelOptions  modelOptions = {} 
)
inline

Definition at line 121 of file INetwork.hpp.

References armnn::ValidateOnly.

122  {})
123  : m_ReduceFp32ToFp16(reduceFp32ToFp16)
124  , m_Debug(debug)
125  , m_ReduceFp32ToBf16(reduceFp32ToBf16)
127  , m_ImportEnabled(importEnabled)
128  , m_ModelOptions(modelOptions)
129  {
131  {
132  throw InvalidArgumentException("BFloat16 and Float16 optimization cannot be enabled at the same time.");
133  }
134  }
ModelOptions m_ModelOptions
Definition: INetwork.hpp:168
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:162
Validate all output shapes.

◆ OptimizerOptions() [3/3]

OptimizerOptions ( bool  reduceFp32ToFp16,
bool  debug,
bool  reduceFp32ToBf16 = false,
ShapeInferenceMethod  shapeInferenceMethod = armnn::ShapeInferenceMethod::ValidateOnly,
bool  importEnabled = false,
ModelOptions  modelOptions = {} 
)
inline

Definition at line 136 of file INetwork.hpp.

138  {})
139  : m_ReduceFp32ToFp16(reduceFp32ToFp16)
140  , m_Debug(debug)
141  , m_ReduceFp32ToBf16(reduceFp32ToBf16)
142  , m_shapeInferenceMethod(shapeInferenceMethod)
143  , m_ImportEnabled(importEnabled)
144  , m_ModelOptions(modelOptions)
145  {
147  {
148  throw InvalidArgumentException("BFloat16 and Float16 optimization cannot be enabled at the same time.");
149  }
150  }
ModelOptions m_ModelOptions
Definition: INetwork.hpp:168
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:162

Member Data Documentation

◆ m_Debug

◆ m_ImportEnabled

bool m_ImportEnabled

◆ m_ModelOptions

◆ m_ReduceFp32ToBf16

◆ m_ReduceFp32ToFp16

◆ m_shapeInferenceMethod

ShapeInferenceMethod m_shapeInferenceMethod

Definition at line 162 of file INetwork.hpp.


The documentation for this struct was generated from the following file: