diff options
Diffstat (limited to 'arm_compute/core/Types.h')
-rw-r--r-- | arm_compute/core/Types.h | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/arm_compute/core/Types.h b/arm_compute/core/Types.h index b6409879bb..711b68f236 100644 --- a/arm_compute/core/Types.h +++ b/arm_compute/core/Types.h @@ -2140,6 +2140,14 @@ public: { return _activation_info; } + /** Set activation layer info + * + * @param[in] activation_info ActivationLayerInfo object to set + */ + void set_activation_info(const ActivationLayerInfo &activation_info) + { + _activation_info = activation_info; + } private: bool _is_a_reshaped; |