#include <Gelu.h>
Definition at line 22 of file Gelu.h.
◆ GeluChef()
| GeluChef::GeluChef |
( |
const tflchef::Operation * |
operation | ) |
|
|
inlineexplicit |
Definition at line 25 of file Gelu.h.
25 : _operation{operation}
26 {
27
28 }
◆ code()
| tflite::BuiltinOperator GeluChef::code |
( |
void |
| ) |
const |
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 31 of file Gelu.h.
31{ return tflite::BuiltinOperator_GELU; }
◆ type()
| tflite::BuiltinOptions GeluChef::type |
( |
void |
| ) |
const |
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 33 of file Gelu.h.
33{ return tflite::BuiltinOptions_GeluOptions; }
◆ value()
Implements OpChef.
Definition at line 19 of file Gelu.cpp.
20{
21 assert(_operation->has_gelu_options());
22
23 const auto &
options = _operation->gelu_options();
24
25 tflite::GeluOptionsBuilder options_builder{fbb};
26 options_builder.add_approximate(
options.approximate());
27
28 return options_builder.
Finish().Union();
29}
void Finish(Offset< T > root, const char *file_identifier=nullptr)
Finish serializing a buffer by writing the root offset.
References flatbuffers::FlatBufferBuilderImpl< Is64Aware >::Finish().
The documentation for this class was generated from the following files:
- compiler/tflchef/core/src/Op/Gelu.h
- compiler/tflchef/core/src/Op/Gelu.cpp