19 #include "ngraph/op/op.hpp"
32 static constexpr NodeTypeInfo type_info{
"GatherTree", 1};
33 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
43 const Output<Node>& parent_idx,
44 const Output<Node>& max_seq_len,
45 const Output<Node>& end_token);
50 virtual std::shared_ptr<Node>
51 clone_with_new_inputs(
const OutputVector& new_args)
const override;
GatherTree(const Output< Node > &step_ids, const Output< Node > &parent_idx, const Output< Node > &max_seq_len, const Output< Node > &end_token)
const NodeTypeInfo & get_type_info() const override
Definition: gather_tree.hpp:33
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28
Visits the attributes of a node, primarily for serialization-like tasks.
Definition: attribute_visitor.hpp:70
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Generates the complete beams from the ids per each step and the parent beam ids.
Definition: gather_tree.hpp:30
Root of all actual ops.
Definition: op.hpp:29