19 #include "ngraph/coordinate.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/strides.hpp"
34 static constexpr NodeTypeInfo type_info{
"VariadicSplit", 1};
35 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
48 const Output<Node>& axis,
49 const Output<Node>& split_lengths);
54 virtual std::shared_ptr<Node>
55 clone_with_new_inputs(
const OutputVector& new_args)
const override;
57 bool evaluate(
const HostTensorVector& outputs,
58 const HostTensorVector& inputs)
const override;
62 using v1::VariadicSplit;
size_t get_default_output_index() const override
Returns the output of the default output, or throws if there is none.
Definition: variadic_split.hpp:56
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
VariadicSplit()=default
Constructs a variadic split operation.
VariadicSplit operation splits an input tensor into pieces along some axis. The pieces may have varia...
Definition: variadic_split.hpp:32
const NodeTypeInfo & get_type_info() const override
Definition: variadic_split.hpp:35
VariadicSplit(const Output< Node > &data, const Output< Node > &axis, const Output< Node > &split_lengths)
Constructs a variadic split operation.
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
Root of all actual ops.
Definition: op.hpp:29