diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2017-10-04 16:53:58 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | ff421f2100e0e9e532f5fe78585300546af61690 (patch) | |
tree | 9ba5a1bfe64b5b10f70c64a965f9c5ca14de9ce3 /arm_compute/graph/Graph.h | |
parent | 925ca0f7402115da3bffb21c04fca0bc822c9b38 (diff) | |
download | ComputeLibrary-ff421f2100e0e9e532f5fe78585300546af61690.tar.gz |
COMPMID-601: Add GraphContext
GraphContext hold all the information about the hints that need to be
passed in the nodes. As these might expand, it serves as a centralized
class for such information.
Change-Id: I0b5527630fb97cc5fa500db0bac8307ff2ea36e6
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/90300
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'arm_compute/graph/Graph.h')
-rw-r--r-- | arm_compute/graph/Graph.h | 30 |
1 files changed, 17 insertions, 13 deletions
diff --git a/arm_compute/graph/Graph.h b/arm_compute/graph/Graph.h index 3c263c2bdd..da41548119 100644 --- a/arm_compute/graph/Graph.h +++ b/arm_compute/graph/Graph.h @@ -65,25 +65,21 @@ public: * @param[in] tensor Tensor to add */ void add_tensor(std::unique_ptr<Tensor> tensor); - /** Sets an execution hint to the graph - * - * @note Hint is propagated to the following node and as per name - * its just a hint/preference to be considered by the graph executor - * - * @param[in] hint execution hint - */ - void set_hint(Hint hint); /** Manually sets the output of the current node * * @param[in] tmp Output info to set */ void set_temp(TensorInfo &&tmp); - /** Sets whether to enable information print out * * @param[in] is_enabled Set to true if need info printed out */ void set_info_enablement(bool is_enabled); + /** Returns the graph hints that are currently used + * + * @return Graph hints + */ + GraphHints &hints(); private: class Private; @@ -106,14 +102,22 @@ Graph &operator<<(Graph &graph, TensorInfo &&info); * @return Updated graph */ Graph &operator<<(Graph &graph, Tensor &&tensor); -/** Overloaded stream operator to provide an execution hint to the graph +/** Overloaded stream operator to provide a target hint to the graph + * + * @param[in, out] graph Graph to provide the hint to + * @param[in] target_hint Target hint to be considered + * + * @return Updated graph + */ +Graph &operator<<(Graph &graph, TargetHint target_hint); +/** Overloaded stream operator to provide a convolution method hint to the graph * - * @param[in, out] graph Graph to provide the hint to - * @param[in] hint Execution hint to be considered + * @param[in, out] graph Graph to provide the hint to + * @param[in] conv_method_hint Convolution method hint to be considered * * @return Updated graph */ -Graph &operator<<(Graph &graph, Hint hint); +Graph &operator<<(Graph &graph, ConvolutionMethodHint conv_method_hint); /** Overloaded stream operator to add a node to the graph * * @param[in, out] graph Graph to add the tensor |