ArmNN
 20.08
Public Member Functions | Public Attributes | List of all members
OptimizerOptions Struct Reference

#include <INetwork.hpp>

Public Member Functions

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

Public Attributes

bool m_ReduceFp32ToFp16
 
bool m_Debug
 
bool m_ReduceFp32ToBf16
 
ShapeInferenceMethod m_shapeInferenceMethod
 
bool m_ImportEnabled
 

Detailed Description

Definition at line 609 of file INetwork.hpp.

Constructor & Destructor Documentation

◆ OptimizerOptions() [1/3]

OptimizerOptions ( )
inline

Definition at line 611 of file INetwork.hpp.

612  : m_ReduceFp32ToFp16(false)
613  , m_Debug(false)
614  , m_ReduceFp32ToBf16(false)
615  , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly)
616  , m_ImportEnabled(false)
617  {}
armnn::OptimizerOptions::m_shapeInferenceMethod
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:657
armnn::ShapeInferenceMethod::ValidateOnly
Validate all output shapes.

◆ OptimizerOptions() [2/3]

OptimizerOptions ( bool  reduceFp32ToFp16,
bool  debug,
bool  reduceFp32ToBf16,
bool  importEnabled 
)
inline

Definition at line 619 of file INetwork.hpp.

620  : m_ReduceFp32ToFp16(reduceFp32ToFp16)
621  , m_Debug(debug)
622  , m_ReduceFp32ToBf16(reduceFp32ToBf16)
623  , m_shapeInferenceMethod(armnn::ShapeInferenceMethod::ValidateOnly)
624  , m_ImportEnabled(importEnabled)
625  {
626  if (m_ReduceFp32ToFp16 && m_ReduceFp32ToBf16)
627  {
628  throw InvalidArgumentException("BFloat16 and Float16 optimization cannot be enabled at the same time.");
629  }
630  }
armnn::OptimizerOptions::m_shapeInferenceMethod
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:657
armnn::ShapeInferenceMethod::ValidateOnly
Validate all output shapes.

◆ OptimizerOptions() [3/3]

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

Definition at line 632 of file INetwork.hpp.

635  : m_ReduceFp32ToFp16(reduceFp32ToFp16)
636  , m_Debug(debug)
637  , m_ReduceFp32ToBf16(reduceFp32ToBf16)
638  , m_shapeInferenceMethod(shapeInferenceMethod)
639  , m_ImportEnabled(importEnabled)
640  {
641  if (m_ReduceFp32ToFp16 && m_ReduceFp32ToBf16)
642  {
643  throw InvalidArgumentException("BFloat16 and Float16 optimization cannot be enabled at the same time.");
644  }
645  }
armnn::OptimizerOptions::m_shapeInferenceMethod
ShapeInferenceMethod m_shapeInferenceMethod
Definition: INetwork.hpp:657

Member Data Documentation

◆ m_Debug

bool m_Debug

Definition at line 651 of file INetwork.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), InferenceModel< IParser, TDataType >::InferenceModel(), and armnn::Optimize().

◆ m_ImportEnabled

bool m_ImportEnabled

Definition at line 660 of file INetwork.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), and armnn::Optimize().

◆ m_ReduceFp32ToBf16

bool m_ReduceFp32ToBf16

Definition at line 654 of file INetwork.hpp.

Referenced by InferenceModel< IParser, TDataType >::InferenceModel(), and armnn::Optimize().

◆ m_ReduceFp32ToFp16

bool m_ReduceFp32ToFp16

Definition at line 648 of file INetwork.hpp.

Referenced by BOOST_AUTO_TEST_CASE(), InferenceModel< IParser, TDataType >::InferenceModel(), and armnn::Optimize().

◆ m_shapeInferenceMethod

ShapeInferenceMethod m_shapeInferenceMethod

Definition at line 657 of file INetwork.hpp.

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