COMPMID-1035 - Add ResneXt50 as a graph example

Change-Id: I42f0e7dab38e45b5eecfe6858eaecee8939c8585
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/129291
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>

1 files changed, 18 insertions, 2 deletions

diff --git a/arm_compute/graph/nodes/EltwiseLayerNode.h b/arm_compute/graph/nodes/EltwiseLayerNode.h
index 5b9fa84bbb..09cbc75b80 100644
--- a/arm_compute/graph/nodes/EltwiseLayerNode.h
+++ b/arm_compute/graph/nodes/EltwiseLayerNode.h
@@ -36,15 +36,29 @@ class EltwiseLayerNode final : public INode
 public:
     /** Constructor
      *
-     * @param[in] op Element-wise operation to perform
+     * @param[in] op       Element-wise operation to perform
+     * @param[in] c_policy (Optional) Convert policy used for the operation
+     * @param[in] r_policy (Optional) Rounding policy used for the operation
      */
-    EltwiseLayerNode(EltwiseOperation op);
+    EltwiseLayerNode(EltwiseOperation op, ConvertPolicy c_policy = ConvertPolicy::SATURATE, RoundingPolicy r_policy = RoundingPolicy::TO_ZERO);
     /** Eltwise operation accessor
      *
      * @return Eltwise operation that is to be performed by the node
      */
     EltwiseOperation eltwise_operation() const;
 
+    /** Convert policy accessor
+     *
+     * @return Convert policy that is used in the node
+     */
+    ConvertPolicy convert_policy() const;
+
+    /** Rounding policy accessor
+     *
+     * @return Convert policy that is used in the node
+     */
+    RoundingPolicy rounding_policy() const;
+
     // Inherited overridden methods:
     NodeType         type() const override;
     bool             forward_descriptors() override;
@@ -53,6 +67,8 @@ public:
 
 private:
     EltwiseOperation _op;
+    ConvertPolicy    _convert_policy;
+    RoundingPolicy   _rounding_policy;
 };
 } // namespace graph
 } // namespace arm_compute