19 #include "ngraph/op/op.hpp"
28 class NGRAPH_DEPRECATED(
29 "This operation is deprecated and will be removed soon. "
30 "Use v1::Gather instead of it.") NGRAPH_API Gather :
public Op
32 NGRAPH_SUPPRESS_DEPRECATED_START
34 static constexpr NodeTypeInfo type_info{
"Gather", 0};
35 const NodeTypeInfo& get_type_info()
const override {
return type_info; }
40 Gather(
const Output<Node>& params,
const Output<Node>& indices,
size_t axis = 0);
42 void validate_and_infer_types()
override;
44 size_t get_axis()
const {
return m_axis; }
45 void set_axis(
size_t axis) { m_axis = axis; }
46 virtual std::shared_ptr<Node>
47 clone_with_new_inputs(
const OutputVector& new_args)
const override;
48 bool evaluate(
const HostTensorVector& outputs,
49 const HostTensorVector& inputs)
const override;
53 NGRAPH_SUPPRESS_DEPRECATED_END
63 static const int64_t AXIS_NOT_SET_VALUE = std::numeric_limits<int64_t>::max();
64 static constexpr NodeTypeInfo type_info{
"Gather", 1};
65 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
71 const Output<Node>& indices,
72 const Output<Node>& axis);
75 int64_t get_axis()
const;
79 virtual std::shared_ptr<Node>
80 clone_with_new_inputs(
const OutputVector& new_args)
const override;
82 bool evaluate(
const HostTensorVector& outputs,
83 const HostTensorVector& inputs)
const override;
88 NGRAPH_SUPPRESS_DEPRECATED_START
90 NGRAPH_SUPPRESS_DEPRECATED_END