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

#include <L2Pool2D.h>

Collaboration diagram for L2Pool2DChef:

Public Member Functions

 L2Pool2DChef (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 L2Pool2D.h.

Constructor & Destructor Documentation

◆ L2Pool2DChef()

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

Definition at line 25 of file L2Pool2D.h.

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

Member Function Documentation

◆ code()

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

Implements OpChef.

Definition at line 31 of file L2Pool2D.h.

31{ return tflite::BuiltinOperator_L2_POOL_2D; }

◆ type()

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

Implements OpChef.

Definition at line 33 of file L2Pool2D.h.

33{ return tflite::BuiltinOptions_Pool2DOptions; }

◆ value()

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

Implements OpChef.

Definition at line 22 of file L2Pool2D.cpp.

23{
24 auto &operation = (*_operation);
25
26 assert(operation.has_l2pool2d_options());
27
28 auto options = operation.l2pool2d_options();
29
31 auto tflite_activation = as_tflite_activation(options.activation());
32
33 tflite::Pool2DOptionsBuilder options_builder{fbb};
34 options_builder.add_padding(tflite_padding);
35 options_builder.add_stride_h(options.stride_h());
36 options_builder.add_stride_w(options.stride_w());
37 options_builder.add_filter_width(options.filter_width());
38 options_builder.add_filter_height(options.filter_height());
39 options_builder.add_fused_activation_function(tflite_activation);
40
41 return options_builder.Finish().Union();
42}
void Finish(Offset< T > root, const char *file_identifier=nullptr)
Finish serializing a buffer by writing the root offset.
tflite::PaddingType tflite_padding(const PaddingType type)
tflite::ActivationFunctionType as_tflite_activation(const tflchef::Activation &value)
Definition Convert.cpp:37
tflite::Padding as_tflite_padding(const tflchef::Padding &value)
Definition Convert.cpp:22

References as_tflite_activation(), as_tflite_padding(), and flatbuffers::FlatBufferBuilder::Finish().


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