24 #ifndef __ARM_COMPUTE_GRAPH_NODE_CONTEXT_H__ 25 #define __ARM_COMPUTE_GRAPH_NODE_CONTEXT_H__ 52 : _operation(operation), _target(
TargetHint::
DONT_CARE), _inputs(), _outputs(), _parameters() {};
105 template <
typename T>
121 std::vector<arm_compute::ITensor *> _inputs;
122 std::vector<arm_compute::ITensor *> _outputs;
123 std::map<std::string, std::unique_ptr<NodeParameterBase>> _parameters;
126 template <
typename T>
130 _parameters[name] = support::cpp14::make_unique<NodeParameter<T>>(name,
parameter);
133 template <
typename T>
136 auto it = _parameters.find(name);
NodeContext(OperationType operation)
Default Constructor.
T parameter(std::string name) const
Returns the parameter with the given name.
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Interface for NEON tensor.
This file contains all available output stages for GEMMLowp on OpenCL.
void add_output(arm_compute::ITensor *output)
Adds an output to the context.
size_t num_outputs() const
Returns number of output.
arm_compute::ITensor * output(size_t idx) const
Returns output tensor of a given index.
Template parameter implementation.
void add_input(arm_compute::ITensor *input)
Adds an input tensor to the context.
OperationType
Supported layer operations.
void add_parameter(std::string name, T parameter)
Adds a parameter to the context.
#define ARM_COMPUTE_ERROR_ON_MSG(cond,...)
TargetHint
< Execution hint to the graph executor
arm_compute::ITensor * input(size_t idx) const
Returns input tensor of a given index.
void set_target(TargetHint target)
Sets the execution target of the node.
TargetHint target() const
Returns the execution target of this node.
size_t num_inputs() const
Returns number of inputs.
OperationType operation() const
Returns the operation of this node.