22 #include "ngraph/node.hpp"
23 #include "ngraph/op/util/fused_op.hpp"
25 NGRAPH_SUPPRESS_DEPRECATED_START
34 class NGRAPH_DEPRECATED(
35 "This operation is deprecated and will be removed soon. "
36 "Use v1::Split instead of it.") NGRAPH_API Split :
public ngraph::op::util::FusedOp
38 NGRAPH_SUPPRESS_DEPRECATED_START
40 static constexpr NodeTypeInfo type_info{
"Split", 0};
41 const NodeTypeInfo& get_type_info()
const override {
return type_info; }
50 Split(
const Output<Node>& data,
const Output<Node>& axis,
const size_t num_split);
62 Split(
const Output<Node>& data,
63 const Output<Node>& axis,
64 const std::vector<size_t>& splits);
66 void pre_validate_and_infer_types()
override;
68 virtual OutputVector decompose_op()
const override;
70 virtual std::shared_ptr<Node>
71 clone_with_new_inputs(
const OutputVector& new_args)
const override;
73 size_t get_axis()
const {
return m_axis; }
74 const std::vector<size_t>& get_splits()
const {
return m_splits; }
81 std::vector<size_t> m_splits;
82 NGRAPH_SUPPRESS_DEPRECATED_END
92 static constexpr NodeTypeInfo type_info{
"Split", 1};
93 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
102 Split(
const Output<Node>& data,
const Output<Node>& axis,
const size_t num_splits);
106 virtual std::shared_ptr<Node>
107 clone_with_new_inputs(
const OutputVector& new_args)
const override;
109 size_t get_num_splits()
const {
return m_num_splits; }
110 void set_num_splits(
const size_t num_splits) { m_num_splits = num_splits; }
111 bool evaluate(
const HostTensorVector& outputs,
112 const HostTensorVector& inputs)
const override;
119 NGRAPH_SUPPRESS_DEPRECATED_START
121 NGRAPH_SUPPRESS_DEPRECATED_END
125 NGRAPH_SUPPRESS_DEPRECATED_END