19 #include "ngraph/op/op.hpp"
30 static constexpr NodeTypeInfo type_info{
"ReverseSequence", 0};
31 const NodeTypeInfo&
get_type_info()
const override {
return type_info; }
41 const Output<Node>& seq_lengths,
48 virtual std::shared_ptr<Node>
49 clone_with_new_inputs(
const OutputVector& new_args)
const override;
51 size_t get_batch_axis()
const {
return m_normalized_batch_axis; }
52 int64_t get_origin_batch_axis()
const {
return m_batch_axis; }
53 void set_batch_axis(int64_t batch_axis) { m_batch_axis = batch_axis; }
54 size_t get_sequence_axis()
const {
return m_normalized_seq_axis; }
55 int64_t get_origin_sequence_axis()
const {
return m_seq_axis; }
56 void set_sequence_axis(int64_t sequence_axis) { m_seq_axis = sequence_axis; }
60 size_t m_normalized_batch_axis;
61 size_t m_normalized_seq_axis;
64 using v0::ReverseSequence;
ReverseSequence(const Output< Node > &arg, const Output< Node > &seq_lengths, int64_t batch_axis, int64_t seq_axis)
Constructs a ReverseSequence operation.
Definition: reverse_sequence.hpp:28
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
const NodeTypeInfo & get_type_info() const override
Definition: reverse_sequence.hpp:31
void validate_and_infer_types() override
Verifies that attributes and inputs are consistent and computes output shapes and element types....
Root of all actual ops.
Definition: op.hpp:29