19 #include "ngraph/op/op.hpp"
31 static const int64_t AXIS_NOT_SET_VALUE = std::numeric_limits<int64_t>::max();
43 int64_t get_axis()
const;
47 virtual std::shared_ptr<Node>
48 clone_with_new_inputs(
const OutputVector& new_args)
const override;
51 const HostTensorVector& inputs)
const override;
52 bool evaluate_lower(
const HostTensorVector& outputs)
const override;
53 bool evaluate_upper(
const HostTensorVector& outputs)
const override;
55 bool constant_fold(OutputVector& output_values,
56 const OutputVector& inputs_values)
override;
59 static const int PARAMS;
60 static const int INDICES;
61 static const int AXIS;
63 bool evaluate_gather(
const HostTensorVector& outputs,
64 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
Gather slices from axis of params according to indices.
Definition: gather.hpp:29
const NodeTypeInfo & get_type_info() const override
Definition: gather.hpp:33
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Gather(const Output< Node > ¶ms, const Output< Node > &indices, const Output< Node > &axis)
bool evaluate(const HostTensorVector &outputs, const HostTensorVector &inputs) const override
Evaluates the op on input_values putting results in output_values.
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28