GRUChef Class Reference

#include <GRU.h>

Collaboration diagram for GRUChef:

Public Member Functions
	GRUChef (const circlechef::Operation *operation)
circle::BuiltinOperator	code (void) const override
circle::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 GRU.h.

Constructor & Destructor Documentation

GRUChef()

GRUChef::GRUChef ( const circlechef::Operation *  operation )

inlineexplicit

Definition at line 25 of file GRU.h.

                                                         : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

circle::BuiltinOperator GRUChef::code ( void  ) const

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file GRU.h.

{ return circle::BuiltinOperator_GRU; }

type()

circle::BuiltinOptions GRUChef::type ( void  ) const

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file GRU.h.

{ return circle::BuiltinOptions_GRUOptions; }

value()

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

overridevirtual

Implements OpChef.

Definition at line 21 of file GRU.cpp.

{
  auto &operation = (*_operation);
 
  assert(operation.has_gru_options());
  auto circle_activation = as_circle_activation(operation.gru_options().activation());
  auto return_sequences = operation.gru_options().return_sequences();
  auto time_major = operation.gru_options().time_major();
 
  circle::GRUOptionsBuilder options_builder{fbb};
  options_builder.add_fused_activation_function(circle_activation);
  options_builder.add_return_sequences(return_sequences);
  options_builder.add_time_major(time_major);
 
  return options_builder.Finish().Union();
}

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

---

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

• compiler/circlechef/core/src/Op/GRU.h
• compiler/circlechef/core/src/Op/GRU.cpp