diff options
author | Isabella Gottardi <isabella.gottardi@arm.com> | 2018-04-06 12:24:55 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:52:35 +0000 |
commit | 88d5b22eb5574d8b564474df2c758d222b3b5547 (patch) | |
tree | 92edf8ecc38a9349faf1ef958998abddcf5b9a8c /arm_compute/graph/nodes/EltwiseLayerNode.h | |
parent | bcedf513938fca9e33331bdef975f0488288bad4 (diff) | |
download | ComputeLibrary-88d5b22eb5574d8b564474df2c758d222b3b5547.tar.gz |
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>
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 |