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

#include <UnidirectionalSequenceLSTM.h>

Collaboration diagram for UnidirectionalSequenceLSTMChef:

Public Member Functions

 UnidirectionalSequenceLSTMChef (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 UnidirectionalSequenceLSTM.h.

Constructor & Destructor Documentation

◆ UnidirectionalSequenceLSTMChef()

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

Definition at line 25 of file UnidirectionalSequenceLSTM.h.

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

Member Function Documentation

◆ code()

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

Implements OpChef.

Definition at line 32 of file UnidirectionalSequenceLSTM.h.

33 {
34 return tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_LSTM;
35 }

◆ type()

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

Implements OpChef.

Definition at line 37 of file UnidirectionalSequenceLSTM.h.

38 {
39 return tflite::BuiltinOptions_UnidirectionalSequenceLSTMOptions;
40 }

◆ value()

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

Implements OpChef.

Definition at line 23 of file UnidirectionalSequenceLSTM.cpp.

24{
25 auto &operation = (*_operation);
26
27 assert(operation.has_unidirectional_sequence_lstm_options());
28
29 tflite::UnidirectionalSequenceLSTMOptionsBuilder options_builder(fbb);
30 options_builder.add_fused_activation_function(
31 as_tflite_activation(operation.unidirectional_sequence_lstm_options().activation()));
32 options_builder.add_cell_clip(operation.unidirectional_sequence_lstm_options().cell_clip());
33 options_builder.add_proj_clip(operation.unidirectional_sequence_lstm_options().proj_clip());
34 options_builder.add_time_major(operation.unidirectional_sequence_lstm_options().time_major());
35 options_builder.add_asymmetric_quantize_inputs(
36 operation.unidirectional_sequence_lstm_options().asymmetric_quantize_inputs());
37
38 return options_builder.Finish().Union();
39}
tflite::ActivationFunctionType as_tflite_activation(const tflchef::Activation &value)
Definition Convert.cpp:37

References as_tflite_activation().


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