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; }
64 bool evaluate(
const HostTensorVector& outputs,
65 const HostTensorVector& inputs)
const override;
71 int64_t m_concat_axis = -1;
Concatenation operation.
Definition: concat.hpp:31
int64_t m_axis
\ brief m_axis stores default value for all iterations
Definition: concat.hpp:69
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
int64_t get_axis() const
Definition: concat.hpp:62
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
Concat()=default
Constructs a concatenation operation.
Concat(const NodeVector &args, int64_t axis)
Constructs a concatenation operation.
Concat(const OutputVector &args, int64_t axis)
Constructs a concatenation operation.
int64_t get_concatenation_axis() const
Definition: concat.hpp:56
Root of all actual ops.
Definition: op.hpp:29