aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/graph/Graph.h
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2017-10-04 16:53:58 +0100
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:35:24 +0000
commitff421f2100e0e9e532f5fe78585300546af61690 (patch)
tree9ba5a1bfe64b5b10f70c64a965f9c5ca14de9ce3 /arm_compute/graph/Graph.h
parent925ca0f7402115da3bffb21c04fca0bc822c9b38 (diff)
downloadComputeLibrary-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.h30
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