diff options
Diffstat (limited to 'arm_compute/graph/nodes/EltwiseLayerNode.h')
| -rw-r--r-- | arm_compute/graph/nodes/EltwiseLayerNode.h | 20 |
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 |