#include <If.h>
Definition at line 22 of file If.h.
◆ IfChef()
| IfChef::IfChef |
( |
const tflchef::Operation * |
operation | ) |
|
|
inlineexplicit |
Definition at line 25 of file If.h.
25 : _operation{operation}
26 {
27
28 }
◆ code()
| tflite::BuiltinOperator IfChef::code |
( |
void |
| ) |
const |
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 31 of file If.h.
31{ return tflite::BuiltinOperator_IF; }
◆ type()
| tflite::BuiltinOptions IfChef::type |
( |
void |
| ) |
const |
|
inlineoverridevirtual |
Implements OpChef.
Definition at line 33 of file If.h.
33{ return tflite::BuiltinOptions_IfOptions; }
◆ value()
Implements OpChef.
Definition at line 19 of file If.cpp.
20{
21 auto &operation = (*_operation);
22
23 assert(operation.has_if_options());
24
25 tflite::IfOptionsBuilder if_options_builder{fbb};
26 if_options_builder.add_then_subgraph_index(operation.if_options().then_subgraph_index());
27 if_options_builder.add_else_subgraph_index(operation.if_options().else_subgraph_index());
28
29 return if_options_builder.
Finish().Union();
30}
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/If.h
- compiler/tflchef/core/src/Op/If.cpp