ArmNN
 20.02
BiasAndWeightsTypesCompatible Struct Reference

#include <LayerSupportRules.hpp>

Inheritance diagram for BiasAndWeightsTypesCompatible:
Rule

Public Member Functions

template<typename Container >
 BiasAndWeightsTypesCompatible (const TensorInfo &info, const Container &c)
 
- Public Member Functions inherited from Rule
bool operator() () const
 

Additional Inherited Members

- Public Attributes inherited from Rule
bool m_Res = true
 

Detailed Description

Definition at line 125 of file LayerSupportRules.hpp.

Constructor & Destructor Documentation

◆ BiasAndWeightsTypesCompatible()

BiasAndWeightsTypesCompatible ( const TensorInfo info,
const Container &  c 
)
inline

Definition at line 128 of file LayerSupportRules.hpp.

References armnn::GetBiasTypeFromWeightsType(), TensorInfo::GetDataType(), armnn::info, and Rule::m_Res.

129  {
130  m_Res = std::any_of(c.begin(), c.end(), [&info](DataType dt)
131  {
132  return dt == GetBiasTypeFromWeightsType(info.GetDataType()).value();
133  });
134  }
armnn::Optional< armnn::DataType > GetBiasTypeFromWeightsType(armnn::Optional< armnn::DataType > weightsType)
DataType
Definition: Types.hpp:32

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