SqueezeChef Class Reference

#include <Squeeze.h>

Collaboration diagram for SqueezeChef:

Public Member Functions
	SqueezeChef (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 Squeeze.h.

Constructor & Destructor Documentation

SqueezeChef()

SqueezeChef::SqueezeChef ( const tflchef::Operation *  operation )

inlineexplicit

Definition at line 25 of file Squeeze.h.

                                                          : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

tflite::BuiltinOperator SqueezeChef::code ( void  ) const

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file Squeeze.h.

{ return tflite::BuiltinOperator_SQUEEZE; }

type()

tflite::BuiltinOptions SqueezeChef::type ( void  ) const

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file Squeeze.h.

{ return tflite::BuiltinOptions_SqueezeOptions; }

value()

flatbuffers::Offset< void > SqueezeChef::value ( flatbuffers::FlatBufferBuilder &  fbb ) const

overridevirtual

Implements OpChef.

Definition at line 23 of file Squeeze.cpp.

{
  auto &operation = (*_operation);
 
  assert(operation.has_squeeze_options());
 
  const auto &options = operation.squeeze_options();
  // Note: 'CreateVector' should be placed before 'CreateOptions'
  //       Read flatbuffers.h 'void NotNested()' for more information
  auto fb_squeeze_dims =
    fbb.CreateVector(options.squeeze_dim().data(), options.squeeze_dim().size());
 
  return tflite::CreateSqueezeOptions(fbb, fb_squeeze_dims).Union();
}

References flatbuffers::FlatBufferBuilder::CreateVector().

---

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

• compiler/tflchef/core/src/Op/Squeeze.h
• compiler/tflchef/core/src/Op/Squeeze.cpp