ONE - On-device Neural Engine
Loading...
Searching...
No Matches
ReverseSequenceChef Class Referencefinal

#include <ReverseSequence.h>

Collaboration diagram for ReverseSequenceChef:

Public Member Functions

 ReverseSequenceChef (const tflchef::Operation *operation)
 
tflite::BuiltinOperator code (void) const override
 
tflite::BuiltinOptions type (void) const override
 
flatbuffers::Offset< void > value (flatbuffers::FlatBufferBuilder &fbb) const override
 
- Public Member Functions inherited from OpChef
virtual ~OpChef ()=default
 
virtual flatbuffers::Offset< flatbuffers::Vector< uint8_t > > custom_value (flatbuffers::FlatBufferBuilder &fbb) const
 
virtual ~OpChef ()=default
 
virtual flatbuffers::Offset< flatbuffers::Vector< uint8_t > > custom_value (flatbuffers::FlatBufferBuilder &fbb) const
 

Detailed Description

Definition at line 22 of file ReverseSequence.h.

Constructor & Destructor Documentation

◆ ReverseSequenceChef()

ReverseSequenceChef::ReverseSequenceChef ( const tflchef::Operation *  operation)
inlineexplicit

Definition at line 25 of file ReverseSequence.h.

25 : _operation{operation}
26 {
27 // DO NOTHING
28 }

Member Function Documentation

◆ code()

tflite::BuiltinOperator ReverseSequenceChef::code ( void  ) const
inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file ReverseSequence.h.

32 {
33 return tflite::BuiltinOperator_REVERSE_SEQUENCE;
34 }

◆ type()

tflite::BuiltinOptions ReverseSequenceChef::type ( void  ) const
inlineoverridevirtual

Implements OpChef.

Definition at line 36 of file ReverseSequence.h.

37 {
38 return tflite::BuiltinOptions_ReverseSequenceOptions;
39 }

◆ value()

flatbuffers::Offset< void > ReverseSequenceChef::value ( flatbuffers::FlatBufferBuilder fbb) const
overridevirtual

Implements OpChef.

Definition at line 19 of file ReverseSequence.cpp.

20{
21 auto &operation = (*_operation);
22
23 assert(operation.has_reverse_sequence_options());
24
25 auto options = operation.reverse_sequence_options();
26
27 auto tflite_seq_dim = options.seq_dim();
28 auto tflite_batch_dim = options.batch_dim();
29
30 tflite::ReverseSequenceOptionsBuilder options_builder{fbb};
31
32 options_builder.add_seq_dim(tflite_seq_dim);
33 options_builder.add_batch_dim(tflite_batch_dim);
34
35 return options_builder.Finish().Union();
36}
void Finish(Offset< T > root, const char *file_identifier=nullptr)
Finish serializing a buffer by writing the root offset.

References flatbuffers::FlatBufferBuilder::Finish().


The documentation for this class was generated from the following files: