22 #include "ngraph/node.hpp"
23 #include "ngraph/op/util/fused_op.hpp"
35 NGRAPH_RTTI_DECLARATION;
49 virtual std::shared_ptr<Node>
50 clone_with_new_inputs(
const OutputVector& new_args)
const override;
52 size_t get_num_splits()
const {
return m_num_splits; }
53 void set_num_splits(
const size_t num_splits) { m_num_splits = num_splits; }
55 const HostTensorVector& inputs)
const override;
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
Splits the input tensor into a list of equal sized tensors.
Definition: split.hpp:33
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
Split()=default
Constructs a split operation.
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Split(const Output< Node > &data, const Output< Node > &axis, const size_t num_splits)
Constructs a split operation.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28