ArmNN
 21.02
BiasAndWeightsTypesMatch Struct Reference

#include <LayerSupportRules.hpp>

Inheritance diagram for BiasAndWeightsTypesMatch:
Rule

Public Member Functions

 BiasAndWeightsTypesMatch (const TensorInfo &biases, const TensorInfo &weights)
 
- 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 118 of file LayerSupportRules.hpp.

Constructor & Destructor Documentation

◆ BiasAndWeightsTypesMatch()

BiasAndWeightsTypesMatch ( const TensorInfo biases,
const TensorInfo weights 
)
inline

Definition at line 120 of file LayerSupportRules.hpp.

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

121  {
122  m_Res = biases.GetDataType() == GetBiasTypeFromWeightsType(weights.GetDataType()).value();
123  }
armnn::Optional< armnn::DataType > GetBiasTypeFromWeightsType(armnn::Optional< armnn::DataType > weightsType)

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