24 #ifndef ARM_COMPUTE_GRAPH_INODE_H 25 #define ARM_COMPUTE_GRAPH_INODE_H 51 virtual ~INode() =
default;
91 std::string
name()
const;
148 const std::vector<TensorID> &
inputs()
const;
153 const std::vector<TensorID> &
outputs()
const;
251 std::vector<TensorID> _outputs;
252 std::vector<EdgeID> _input_edges;
253 std::set<EdgeID> _output_edges;
const std::vector< TensorID > & inputs() const
Returns inputs of the node.
Edge * input_edge(size_t idx) const
Returns the edge of a given input of the node.
void set_assigned_target(Target target)
Sets the final execution target.
std::string name() const
Returns node's name.
virtual ~INode()=default
Destructor.
const std::set< EdgeID > & output_edges() const
Returns output edge set.
virtual void accept(INodeVisitor &v)=0
Accepts a node visitor.
Target assigned_target() const
Returns assigned target for this node.
size_t num_outputs() const
Returns number of outputs of the node.
const Graph * graph() const
Returns node's Graph.
virtual TensorDescriptor configure_output(size_t idx) const =0
Calculates output configuration.
INode & operator=(const INode &)=delete
Prevent instances of this class from being copy assigned (As this class contains pointers) ...
EdgeID input_edge_id(size_t idx) const
Returns the edge ID of a given input of the node.
Copyright (c) 2017-2021 Arm Limited.
void set_requested_target(Target target)
Sets target preference.
void set_graph(Graph *g)
Sets the graph that this node is registered to.
const std::vector< EdgeID > & input_edges() const
Returns input edge set.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
Target requested_target() const
Returns requested target for this node.
NodeID id() const
Returns node's ID.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
const std::vector< TensorID > & outputs() const
Returns outputs of the node.
void set_id(NodeID id)
Sets the node id.
void set_common_node_parameters(NodeParams common_params)
Sets common node parameters.
NodeParams common_node_params() const
Returns common node parameters.
virtual Status validate() const
Validate node.
size_t num_inputs() const
Returns number of inputs of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
void set_output_tensor(TensorID tid, size_t idx)
Sets the output tensor of at a given index.
virtual bool forward_descriptors()=0
Forwards descriptor information to outputs if possible.
virtual NodeType type() const =0
Returns node's type.
Tensor * input(size_t idx) const
Returns the tensor of a given input of the node.