BatchMatMulV2Chef Class Reference

#include <BatchMatMulV2.h>

Collaboration diagram for BatchMatMulV2Chef:

Public Member Functions
	BatchMatMulV2Chef (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
flatbuffers::Offset< flatbuffers::Vector< uint8_t > >	custom_value (flatbuffers::FlatBufferBuilder &fbb) const override
Public Member Functions inherited from OpChef
virtual	~OpChef ()=default
virtual	~OpChef ()=default

Detailed Description

Definition at line 22 of file BatchMatMulV2.h.

Constructor & Destructor Documentation

BatchMatMulV2Chef()

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

inlineexplicit

Definition at line 25 of file BatchMatMulV2.h.

                                                                : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file BatchMatMulV2.h.

{ return tflite::BuiltinOperator_CUSTOM; }

custom_value()

flatbuffers::Offset< flatbuffers::Vector< uint8_t > > BatchMatMulV2Chef::custom_value ( flatbuffers::FlatBufferBuilder &  fbb ) const

overridevirtual

REGISTER_OP("BatchMatMulV2") .Input("x: T") .Input("y: T") .Output("output: T") .Attr( "T: {bfloat16, half, float, double, int32, int64, complex64, " "complex128}") .Attr("adj_x: bool = false") .Attr("adj_y: bool = false") .SetShapeFn(shape_inference::BatchMatMulV2Shape);

Reimplemented from OpChef.

Definition at line 29 of file BatchMatMulV2.cpp.

{
  auto &operation = (*_operation);
 
  check_custom_op_value(operation, "BatchMatMulV2");
 
  auto flex_buffers = std::make_unique<flexbuffers::Builder>();
  size_t map_start = flex_buffers->StartMap();
 
  flex_buffers->Bool("adj_x", operation.batch_matmul_options().adj_x());
  flex_buffers->Bool("adj_y", operation.batch_matmul_options().adj_y());
  // TODO Support more data types
  flex_buffers->Int("T", tflite::TensorType_FLOAT32);
 
  flex_buffers->EndMap(map_start);
  flex_buffers->Finish();
 
  auto circle_custom_options = fbb.CreateVector(flex_buffers->GetBuffer());
  return circle_custom_options;
}

References check_custom_op_value(), and flatbuffers::FlatBufferBuilder::CreateVector().

type()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file BatchMatMulV2.h.

{ return tflite::BuiltinOptions_NONE; }

value()

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

overridevirtual

Implements OpChef.

Definition at line 23 of file BatchMatMulV2.cpp.

{
  return flatbuffers::Offset<void>();
}

---

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

• compiler/tflchef/core/src/CustomOp/BatchMatMulV2.h
• compiler/tflchef/core/src/CustomOp/BatchMatMulV2.cpp