TransposeConvChef Class Reference

#include <TransposeConv.h>

Collaboration diagram for TransposeConvChef:

Public Member Functions
	TransposeConvChef (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 TransposeConv.h.

Constructor & Destructor Documentation

TransposeConvChef()

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

inlineexplicit

Definition at line 25 of file TransposeConv.h.

                                                                : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file TransposeConv.h.

  {
    return tflite::BuiltinOperator_TRANSPOSE_CONV;
  }

type()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 36 of file TransposeConv.h.

  {
    return tflite::BuiltinOptions_TransposeConvOptions;
  }

value()

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

overridevirtual

Implements OpChef.

Definition at line 22 of file TransposeConv.cpp.

{
  auto &operation = (*_operation);
 
  tflite::TransposeConvOptionsBuilder options_builder{fbb};
 
  assert(operation.has_transpose_conv_options());
 
  auto tflite_padding = as_tflite_padding(operation.transpose_conv_options().padding());
 
  options_builder.add_padding(tflite_padding);
 
  options_builder.add_stride_h(operation.transpose_conv_options().stride_h());
  options_builder.add_stride_w(operation.transpose_conv_options().stride_w());
 
  // TODO remove calling has_activation
  auto chef_activation = operation.transpose_conv_options().has_activation()
                           ? operation.transpose_conv_options().activation()
                           : tflchef::NONE;
  auto tflite_activation = as_tflite_activation(chef_activation);
  options_builder.add_fused_activation_function(tflite_activation);
 
  return options_builder.Finish().Union();
}

References as_tflite_activation(), and as_tflite_padding().

---

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

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