From 2a2db590fd179dcb8e1a575293cd2b887e2dc246 Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 15 Aug 2018 12:14:46 +0100 Subject: COMPMID-1505: Add native grouping support at graph level Change-Id: Iedc91b0aee743b59af5140c8acb8124548da3163 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/144362 Tested-by: Jenkins Reviewed-by: Giorgio Arena Reviewed-by: Michele DiGiorgio --- arm_compute/graph/Utils.h | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'arm_compute/graph/Utils.h') diff --git a/arm_compute/graph/Utils.h b/arm_compute/graph/Utils.h index 3604bad4af..1a0509b6fa 100644 --- a/arm_compute/graph/Utils.h +++ b/arm_compute/graph/Utils.h @@ -115,6 +115,18 @@ size_t get_dimension_size(const TensorDescriptor &descriptor, const DataLayoutDi * @return Idx of given dimension */ size_t get_dimension_idx(const TensorDescriptor &descriptor, const DataLayoutDimension data_layout_dimension); +/** Get the list of driving nodes of a given node + * + * @param[in] node Node to find the driving node of + * + * @return A list with the driving node of a given node + */ +std::vector get_driving_nodes(const INode &node); +/** Configures tensor + * + * @param[in, out] tensor Tensor to configure + */ +void configure_tensor(Tensor *tensor); } // namespace graph } // namespace arm_compute #endif /* __ARM_COMPUTE_GRAPH_UTILS_H__ */ -- cgit v1.2.1