24 #ifndef ARM_COMPUTE_GRAPH_UTILS_H 25 #define ARM_COMPUTE_GRAPH_UTILS_H 40 return utility_node_types.find(node->
type()) != utility_node_types.end();
TensorDescriptor get_tensor_descriptor(const Graph &g, TensorID tid)
Returns the tensor descriptor of a given tensor.
void force_target_to_graph(Graph &g, Target target)
Forces a single target to all graph constructs.
Graph configuration structure Device target types.
void configure_tensor(Tensor *tensor)
Configures tensor.
std::vector< NodeIdxPair > get_driving_nodes(const INode &node)
Get the list of driving nodes of a given node.
size_t get_dimension_size(const TensorDescriptor &descriptor, const DataLayoutDimension data_layout_dimension)
Get size of a tensor's given dimension depending on its layout.
DataLayoutDimension
[DataLayout enum definition]
bool is_utility_node(INode *node)
const DataLayout data_layout
#define ARM_COMPUTE_RETURN_ERROR_ON(cond)
If the condition is true, an error is returned.
void set_accessor(std::unique_ptr< ITensorAccessor > accessor)
Sets the backend tensor accessor.
Copyright (c) 2017-2021 Arm Limited.
Status set_tensor_accessor(Tensor *tensor, std::unique_ptr< ITensorAccessor > accessor)
Sets an accessor on a given tensor.
TensorDescriptor & desc()
TensorInfo metadata accessor.
Target get_default_target()
Returns default target for execution.
void setup_requested_backend_context(GraphContext &ctx, Target target)
Setups requested backend context if it exists, is supported and hasn't been initialized already...
bool is_target_supported(Target target)
Checks if a specific target is supported.
PassManager create_default_pass_manager(Target target, const GraphConfig &cfg)
Creates a default PassManager.
virtual NodeType type() const =0
Returns node's type.
DataLayout
[DataLayout enum definition]
void release_default_graph_context(GraphContext &ctx)
Default releases the graph context if not done manually.
const Tensor * tensor(TensorID id) const
Get tensor object given its id.
size_t get_dimension_idx(DataLayout data_layout, const DataLayoutDimension data_layout_dimension)
Get index of a tensor's given dimension depending on its layout.