diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-08-15 12:14:46 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:54:54 +0000 |
| commit | 2a2db590fd179dcb8e1a575293cd2b887e2dc246 (patch) | |
| tree | 5e10da7cb6777f3020b84a2389b279ceef2be5ee /arm_compute/graph/Utils.h | |
| parent | c1961b51df2e15a01a5950139e81bbd47fbfa627 (diff) | |
| download | ComputeLibrary-2a2db590fd179dcb8e1a575293cd2b887e2dc246.tar.gz | |
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 <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Michele DiGiorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute/graph/Utils.h')
| -rw-r--r-- | arm_compute/graph/Utils.h | 12 |
1 files changed, 12 insertions, 0 deletions
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<NodeIdxPair> 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__ */ |