21 #include "ngraph/op/op.hpp"
33 NGRAPH_RTTI_DECLARATION;
41 Concat(
const OutputVector& args, int64_t axis);
47 Concat(
const NodeVector& args, int64_t axis);
52 virtual std::shared_ptr<Node>
53 clone_with_new_inputs(
const OutputVector& new_args)
const override;
57 void set_concatenation_axis(int64_t concatenation_axis)
59 m_concat_axis = concatenation_axis;
63 void set_axis(int64_t axis) { m_axis = axis; }
65 const HostTensorVector& inputs)
const override;
66 bool evaluate_lower(
const HostTensorVector& output_values)
const override;
67 bool evaluate_upper(
const HostTensorVector& output_values)
const override;
73 int64_t m_concat_axis = -1;
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
Root of all actual ops.
Definition: op.hpp:29
Concatenation operation.
Definition: concat.hpp:31
int64_t get_axis() const
Definition: concat.hpp:62
int64_t get_concatenation_axis() const
Definition: concat.hpp:56
Concat(const OutputVector &args, int64_t axis)
Constructs a concatenation operation.
int64_t m_axis
\ brief m_axis stores default value for all iterations
Definition: concat.hpp:71
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
Concat()=default
Constructs a concatenation operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Concat(const NodeVector &args, int64_t axis)
Constructs a concatenation operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28