ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
RoPEChef Class Referencefinal

#include <RoPE.h>

Collaboration diagram for RoPEChef:

Public Member Functions

 RoPEChef (const circlechef::Operation *operation)
 
circle::BuiltinOperator code (void) const override
 
circle::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 RoPE.h.

Constructor & Destructor Documentation

◆ RoPEChef()

RoPEChef::RoPEChef ( const circlechef::Operation *  operation)
inlineexplicit

Definition at line 25 of file RoPE.h.

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

Member Function Documentation

◆ code()

circle::BuiltinOperator RoPEChef::code ( void  ) const
inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file RoPE.h.

31{ return circle::BuiltinOperator_ROPE; }

◆ type()

circle::BuiltinOptions RoPEChef::type ( void  ) const
inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file RoPE.h.

33{ return circle::BuiltinOptions_RoPEOptions; }

◆ value()

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

Implements OpChef.

Definition at line 21 of file RoPE.cpp.

22{
23 auto &operation = (*_operation);
24 assert(operation.has_rope_options());
25
26 circle::RoPEOptionsBuilder options_builder{fbb};
27 options_builder.add_mode(static_cast<circle::RoPEMode>(operation.rope_options().mode()));
28
29 return 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::FlatBufferBuilder::Finish().


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