24 #ifndef __ARM_COMPUTE_GRAPH_INODE_H__ 25 #define __ARM_COMPUTE_GRAPH_INODE_H__ 50 virtual ~INode() =
default;
90 std::string
name()
const;
147 const std::vector<TensorID> &
inputs()
const;
152 const std::vector<TensorID> &
outputs()
const;
245 std::vector<TensorID> _outputs;
246 std::vector<EdgeID> _input_edges;
247 std::set<EdgeID> _output_edges;
void set_assigned_target(Target target)
Sets the final execution target.
const std::vector< TensorID > & outputs() const
Returns outputs of the node.
virtual ~INode()=default
Destructor.
EdgeID input_edge_id(size_t idx) const
Returns the edge ID of a given input of the node.
virtual void accept(INodeVisitor &v)=0
Accepts a node visitor.
const std::set< EdgeID > & output_edges() const
Returns output edge set.
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) ...
const Graph * graph() const
Returns node's Graph.
TensorID input_id(size_t idx) const
Returns the tensor ID of a given input of the node.
size_t num_inputs() const
Returns number of inputs of the node.
This file contains all available output stages for GEMMLowp on OpenCL.
const std::vector< TensorID > & inputs() const
Returns inputs of the node.
TensorID output_id(size_t idx) const
Returns the tensor ID of a given output of the node.
void set_requested_target(Target target)
Sets target preference.
void set_graph(Graph *g)
Sets the graph that this node is registered to.
Target assigned_target() const
Returns assigned target for this node.
virtual Status validate() const
Validate node.
std::string name() const
Returns node's name.
Edge * input_edge(size_t idx) const
Returns the edge of a given input of the node.
const std::vector< EdgeID > & input_edges() const
Returns input edge set.
void set_id(NodeID id)
Sets the node id.
void set_common_node_parameters(NodeParams common_params)
Sets common node parameters.
NodeID id() const
Returns node's ID.
Target requested_target() const
Returns requested target for this node.
Tensor * output(size_t idx) const
Returns the tensor of a given output of the node.
Tensor * input(size_t idx) const
Returns the tensor of a given input 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.
size_t num_outputs() const
Returns number of outputs of the node.