ArmNN
 23.11
BiasAndWeightsTypesMatch Struct Reference

#include <LayerSupportRules.hpp>

Inheritance diagram for BiasAndWeightsTypesMatch:
[legend]
Collaboration diagram for BiasAndWeightsTypesMatch:
[legend]

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.

121  {
122  m_Res = biases.GetDataType() == GetBiasTypeFromWeightsType(weights.GetDataType()).value();
123  }

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


The documentation for this struct was generated from the following file:
armnn::Rule::m_Res
bool m_Res
Definition: LayerSupportRules.hpp:55
armnn::GetBiasTypeFromWeightsType
armnn::Optional< armnn::DataType > GetBiasTypeFromWeightsType(armnn::Optional< armnn::DataType > weightsType)
Definition: LayerSupportRules.hpp:14