CastChef Class Reference

#include <Cast.h>

Collaboration diagram for CastChef:

Public Member Functions
	CastChef (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 Cast.h.

Constructor & Destructor Documentation

CastChef()

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

inlineexplicit

Definition at line 25 of file Cast.h.

                                                       : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file Cast.h.

{ return tflite::BuiltinOperator_CAST; }

type()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file Cast.h.

{ return tflite::BuiltinOptions_CastOptions; }

value()

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

overridevirtual

Implements OpChef.

Definition at line 22 of file Cast.cpp.

{
  auto &operation = (*_operation);
 
  if (!operation.has_cast_options())
    return 0;
 
  auto tflite_in_data_type = as_tflite_tensortype(operation.cast_options().in_data_type());
  auto tflite_out_data_type = as_tflite_tensortype(operation.cast_options().out_data_type());
 
  tflite::CastOptionsBuilder cast_options_builder{fbb};
  cast_options_builder.add_in_data_type(tflite_in_data_type);
  cast_options_builder.add_out_data_type(tflite_out_data_type);
 
  return cast_options_builder.Finish().Union();
}

References as_tflite_tensortype(), and flatbuffers::FlatBufferBuilder::Finish().

---

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

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