-DARMNNREF_ENABLED
endif # ARMNN_REF_ENABLED == 1
-# Only enable on Android P
-ifeq (($(P_OR_LATER),1))
- ifeq (($(Q_OR_LATER),0))
-LOCAL_CFLAGS += \
- -DARMNN_MIXED_PRECISION_FP16_POOLING
- endif # Q_OR_LATER=0
-endif # P_OR_LATER=1
-
ifeq ($(Q_OR_LATER),1)
LOCAL_CFLAGS += \
-DBOOST_NO_AUTO_PTR
// enable fp_mixed_precision for the the FP16 cases that
// accumulation reaches a limit beyond which there is no more increment of the value
bool fpMixedPrecision = false;
-#ifdef ARMNN_MIXED_PRECISION_FP16_POOLING
if (input.info()->data_type() == arm_compute::DataType::F16
- && m_Data.m_Parameters.m_PoolType == PoolingAlgorithm::Average)
+ && m_Data.m_Parameters.m_PoolType == PoolingAlgorithm::Average
+ && m_Data.m_Parameters.m_PoolWidth >= 100
+ && input.info()->dimension(3) >= 5
+ && input.info()->dimension(2) * input.info()->dimension(1) >= 3000)
{
fpMixedPrecision = true;
}
-#endif
arm_compute::PoolingLayerInfo layerInfo = BuildArmComputePoolingLayerInfo(m_Data.m_Parameters, fpMixedPrecision);