aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/Types.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r--arm_compute/core/Types.h10
1 files changed, 7 insertions, 3 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h
index 66e1c8ab1f..952c174194 100644
--- a/arm_compute/core/Types.h
+++ b/arm_compute/core/Types.h
@@ -1987,16 +1987,20 @@ enum class WeightFormat
// OHWIo<interleave_by>i<block_by>
inline int interleave_by(const WeightFormat wf)
{
- return ((int)wf >> 8) & 0xFFF;
+ return (static_cast<int>(wf) >> 8) & 0xFFF;
}
inline int block_by(const WeightFormat wf)
{
- return ((int)wf >> 20) & 0xF;
+ return (static_cast<int>(wf) >> 20) & 0xF;
}
-inline bool is_fixed_format(const WeightFormat wf)
+inline bool is_fixed_format(const WeightFormat &wf)
{
return wf != WeightFormat::UNSPECIFIED && wf != WeightFormat::ANY;
}
+inline bool is_fixed_format_fast_math(const WeightFormat &wf)
+{
+ return (static_cast<int>(wf) >> 4) & 0x1;
+}
/** Convolution Layer Weights Information class. This class stores the necessary information to compute convolution layer when the weights are already reshaped */
class WeightsInfo