BatchMatMulChef Class Reference

#include <BatchMatMul.h>

Collaboration diagram for BatchMatMulChef:

Public Member Functions
	BatchMatMulChef (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
	BatchMatMulChef (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 BatchMatMul.h.

Constructor & Destructor Documentation

BatchMatMulChef() [1/2]

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

inlineexplicit

Definition at line 25 of file BatchMatMul.h.

                                                                 : _operation{operation}
  {
    // DO NOTHING
  }

BatchMatMulChef() [2/2]

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

inlineexplicit

Definition at line 25 of file BatchMatMul.h.

                                                              : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code() [1/2]

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file BatchMatMul.h.

{ return circle::BuiltinOperator_BATCH_MATMUL; }

code() [2/2]

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file BatchMatMul.h.

{ return tflite::BuiltinOperator_BATCH_MATMUL; }

type() [1/2]

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

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file BatchMatMul.h.

  {
    return circle::BuiltinOptions_BatchMatMulOptions;
  }

type() [2/2]

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

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file BatchMatMul.h.

  {
    return tflite::BuiltinOptions_BatchMatMulOptions;
  }

value() [1/2]

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

overridevirtual

Implements OpChef.

Definition at line 19 of file BatchMatMul.cpp.

{
  auto &operation = (*_operation);
 
  assert(operation.has_batch_matmul_options());
 
  circle::BatchMatMulOptionsBuilder batch_matmul_options_options_builder{fbb};
  batch_matmul_options_options_builder.add_adjoint_lhs(
    operation.batch_matmul_options().adjoint_lhs());
  batch_matmul_options_options_builder.add_adjoint_rhs(
    operation.batch_matmul_options().adjoint_rhs());
 
  return batch_matmul_options_options_builder.Finish().Union();
}

References flatbuffers::FlatBufferBuilder::Finish().

value() [2/2]

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

overridevirtual

Implements OpChef.

---

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

• compiler/circlechef/core/src/Op/BatchMatMul.h
• compiler/tflchef/core/src/Op/BatchMatMul.h
• compiler/circlechef/core/src/Op/BatchMatMul.cpp
• compiler/tflchef/core/src/Op/BatchMatMul.cpp