19 #include "ngraph/op/op.hpp"
32 static constexpr
NodeTypeInfo type_info{
"GatherTree", 1};
50 virtual std::shared_ptr<Node>
51 clone_with_new_inputs(
const OutputVector& new_args)
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
Generates the complete beams from the ids per each step and the parent beam ids.
Definition: gather_tree.hpp:30
GatherTree(const Output< Node > &step_ids, const Output< Node > &parent_idx, const Output< Node > &max_seq_len, const Output< Node > &end_token)
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
const NodeTypeInfo & get_type_info() const override
Definition: gather_tree.hpp:33
The Intel nGraph C++ API.
Definition: attribute_adapter.hpp:28