diff options
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__ */ |