ArmNN
 22.05.01
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 126 of file LayerSupportRules.hpp.

Constructor & Destructor Documentation

◆ BiasAndWeightsTypesCompatible()

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

Definition at line 129 of file LayerSupportRules.hpp.

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

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

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