19 #include "ngraph/coordinate.hpp"
20 #include "ngraph/op/op.hpp"
21 #include "ngraph/strides.hpp"
34 NGRAPH_RTTI_DECLARATION;
54 virtual std::shared_ptr<Node>
55 clone_with_new_inputs(
const OutputVector& new_args)
const override;
58 const HostTensorVector& inputs)
const override;
61 bool evaluate_variadic_split(
const HostTensorVector& outputs,
62 const HostTensorVector& inputs)
const;
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:71
A handle for one of a node's outputs.
Definition: node_output.hpp:42
Root of all actual ops.
Definition: op.hpp:29
VariadicSplit operation splits an input tensor into pieces along some axis. The pieces may have varia...
Definition: variadic_split.hpp:32
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
VariadicSplit()=default
Constructs a variadic split operation.
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
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