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>
diff --git a/arm_compute/graph/nodes/EltwiseLayerNode.h b/arm_compute/graph/nodes/EltwiseLayerNode.h
index 5b9fa84..09cbc75 100644
--- a/arm_compute/graph/nodes/EltwiseLayerNode.h
+++ b/arm_compute/graph/nodes/EltwiseLayerNode.h
@@ -36,15 +36,29 @@
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 @@
private:
EltwiseOperation _op;
+ ConvertPolicy _convert_policy;
+ RoundingPolicy _rounding_policy;
};
} // namespace graph
} // namespace arm_compute