DepthwiseConv2DChef Class Reference

#include <DepthwiseConv2D.h>

Collaboration diagram for DepthwiseConv2DChef:

Public Member Functions
	DepthwiseConv2DChef (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 DepthwiseConv2D.h.

Constructor & Destructor Documentation

DepthwiseConv2DChef()

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

inlineexplicit

Definition at line 25 of file DepthwiseConv2D.h.

                                                                  : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file DepthwiseConv2D.h.

  {
    return tflite::BuiltinOperator_DEPTHWISE_CONV_2D;
  }

type()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 36 of file DepthwiseConv2D.h.

  {
    return tflite::BuiltinOptions_DepthwiseConv2DOptions;
  }

value()

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

overridevirtual

Implements OpChef.

Definition at line 22 of file DepthwiseConv2D.cpp.

{
  auto &operation = (*_operation);
 
  assert(operation.has_depthwiseconv2d_options());
 
  auto options = operation.depthwiseconv2d_options();
 
  auto tflite_padding = as_tflite_padding(options.padding());
  auto tflite_activation = as_tflite_activation(options.activation());
 
  tflite::DepthwiseConv2DOptionsBuilder options_builder{fbb};
  options_builder.add_padding(tflite_padding);
  options_builder.add_stride_w(options.stride_w());
  options_builder.add_stride_h(options.stride_h());
  options_builder.add_depth_multiplier(options.depth_multiplier());
  options_builder.add_fused_activation_function(tflite_activation);
  options_builder.add_dilation_w_factor(options.dilation_w_factor());
  options_builder.add_dilation_h_factor(options.dilation_h_factor());
 
  return options_builder.Finish().Union();
}

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

---

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

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