ArmNN
 21.08
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

Examples:
CustomMemoryAllocatorSample.cpp.

Definition at line 109 of file INetwork.hpp.

Constructor & Destructor Documentation

◆ OptimizerOptions() [1/3]

OptimizerOptions ( )
inline

Definition at line 111 of file INetwork.hpp.

112  : m_ReduceFp32ToFp16(false)
113  , m_Debug(false)
114  , m_ReduceFp32ToBf16(false)
116  , m_ImportEnabled(false)
117  , m_ModelOptions()
118  {}
ModelOptions m_ModelOptions
Definition: INetwork.hpp:167
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:161
Validate all output shapes.

◆ OptimizerOptions() [2/3]

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

Definition at line 120 of file INetwork.hpp.

References armnn::ValidateOnly.

121  {})
122  : m_ReduceFp32ToFp16(reduceFp32ToFp16)
123  , m_Debug(debug)
124  , m_ReduceFp32ToBf16(reduceFp32ToBf16)
126  , m_ImportEnabled(importEnabled)
127  , m_ModelOptions(modelOptions)
128  {
130  {
131  throw InvalidArgumentException("BFloat16 and Float16 optimization cannot be enabled at the same time.");
132  }
133  }
ModelOptions m_ModelOptions
Definition: INetwork.hpp:167
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:161
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 135 of file INetwork.hpp.

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

Member Data Documentation

◆ m_Debug

◆ m_ImportEnabled

bool m_ImportEnabled

◆ m_ModelOptions

◆ m_ReduceFp32ToBf16

◆ m_ReduceFp32ToFp16

◆ m_shapeInferenceMethod

ShapeInferenceMethod m_shapeInferenceMethod

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