MaxPoolWithArgmaxChef Class Reference

#include <MaxPoolWithArgmax.h>

Collaboration diagram for MaxPoolWithArgmaxChef:

Public Member Functions
	MaxPoolWithArgmaxChef (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 MaxPoolWithArgmax.h.

Constructor & Destructor Documentation

MaxPoolWithArgmaxChef()

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

inlineexplicit

Definition at line 25 of file MaxPoolWithArgmax.h.

                                                                    : _operation{operation}
  {
    // DO NOTHING
  }

Member Function Documentation

code()

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

inlineoverridevirtual

Implements OpChef.

Definition at line 31 of file MaxPoolWithArgmax.h.

{ return tflite::BuiltinOperator_CUSTOM; }

custom_value()

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

overridevirtual

REGISTER_OP("MaxPoolWithArgmax") .Attr("ksize: list(int) >= 4") .Attr("strides: list(int) >= 4") .Attr("Targmax: {int32, int64} = DT_INT64") .Attr(GetPaddingAttrString()) .Attr("include_batch_in_index: bool = false") .Input("input: T") .Output("output: T") .Output("argmax: Targmax") .Attr("T: realnumbertype") .SetShapeFn([](InferenceContext* c) { TF_RETURN_IF_ERROR(shape_inference::MaxPoolShape(c)); c->set_output(1, c->output(0)); return Status::OK(); });

Reimplemented from OpChef.

Definition at line 29 of file MaxPoolWithArgmax.cpp.

{
  auto &operation = (*_operation);
 
  check_custom_op_value(operation, "MaxPoolWithArgmax");
 
  auto flex_buffers = std::make_unique<flexbuffers::Builder>();
  size_t map_start = flex_buffers->StartMap();
 
  auto start = flex_buffers->StartVector("ksize");
  flex_buffers->Add(1);
  flex_buffers->Add(operation.max_pool_with_argmax_options().filter_width());
  flex_buffers->Add(operation.max_pool_with_argmax_options().filter_height());
  flex_buffers->Add(1);
  flex_buffers->EndVector(start, /*typed=*/true, /*fixed=*/false);
  start = flex_buffers->StartVector("strides");
  flex_buffers->Add(1);
  flex_buffers->Add(operation.max_pool_with_argmax_options().stride_w());
  flex_buffers->Add(operation.max_pool_with_argmax_options().stride_h());
  flex_buffers->Add(1);
  flex_buffers->EndVector(start, /*typed=*/true, /*fixed=*/false);
  auto output_type = operation.max_pool_with_argmax_options().output_type();
  assert(output_type == tflchef::INT64 || output_type == tflchef::INT32);
  flex_buffers->Int("Targmax", output_type);
  std::string padding = operation.max_pool_with_argmax_options().padding() ? "VALID" : "SAME";
  flex_buffers->String("padding", padding);
  flex_buffers->Bool("include_batch_in_index",
                     operation.max_pool_with_argmax_options().include_batch_in_index());
  flex_buffers->Int("T", tflchef::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 MaxPoolWithArgmaxChef::type ( void  ) const

inlineoverridevirtual

Implements OpChef.

Definition at line 33 of file MaxPoolWithArgmax.h.

{ return tflite::BuiltinOptions_NONE; }

value()

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

overridevirtual

Implements OpChef.

Definition at line 23 of file MaxPoolWithArgmax.cpp.

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

---

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

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