circlechef::CircleOpBCQGather Class Reference

circlechef operator builder for BCQGather More...

#include <BCQGather.h>

Collaboration diagram for circlechef::CircleOpBCQGather:

Public Member Functions
void	filler (const circle::Operator *op, CircleImport *import, circlechef::ModelRecipe *model_recipe) const override
circlechef::Operation *	build (const circle::Operator *op, CircleImport *import, circlechef::ModelRecipe *model_recipe) const override
Public Member Functions inherited from circlechef::CircleOpChef
virtual	~CircleOpChef ()

Detailed Description

circlechef operator builder for BCQGather

Definition at line 28 of file BCQGather.h.

Member Function Documentation

build()

circlechef::Operation * circlechef::CircleOpBCQGather::build ( const circle::Operator *  op, CircleImport *  import, circlechef::ModelRecipe *  model_recipe  ) const

overridevirtual

Implements circlechef::CircleOpChef.

Definition at line 50 of file BCQGather.cpp.

{
  auto op_params = op->builtin_options_as_BCQGatherOptions();
  assert(op_params != nullptr);
 
  auto operation = model_recipe->add_operation();
 
  operation->set_type("BCQGather");
 
  auto op_options = operation->mutable_bcq_gather_options();
 
  op_options->set_input_hidden_size(op_params->input_hidden_size());
  op_options->set_axis(op_params->axis());
 
  return operation;
}

filler()

void circlechef::CircleOpBCQGather::filler ( const circle::Operator *  op, CircleImport *  import, circlechef::ModelRecipe *  model_recipe  ) const

overridevirtual

Implements circlechef::CircleOpChef.

Definition at line 24 of file BCQGather.cpp.

{
  const std::vector<int32_t> &inputs = as_index_vector(op->inputs());
 
  import->set_tensor_filler(inputs[0]);
 
  const circle::Tensor *tensor1 = import->tensors()->Get(inputs[1]);
  assert(tensor1->type() == circle::TensorType::TensorType_INT32);
  const circle::Buffer *buffer1 = import->buffers()->Get(tensor1->buffer());
  auto vec1 = extract_buffer<int32_t>(buffer1);
  import->set_tensor_filler(inputs[1], vec1);
 
  const circle::Tensor *tensor2 = import->tensors()->Get(inputs[2]);
  assert(tensor2->type() == circle::TensorType::TensorType_INT32);
  const circle::Buffer *buffer2 = import->buffers()->Get(tensor2->buffer());
  auto vec2 = extract_buffer<int32_t>(buffer2);
  import->set_tensor_filler(inputs[2], vec2);
 
  const circle::Tensor *tensor3 = import->tensors()->Get(inputs[3]);
  assert(tensor3->type() == circle::TensorType::TensorType_INT32);
  const circle::Buffer *buffer3 = import->buffers()->Get(tensor3->buffer());
  auto vec3 = extract_buffer<int32_t>(buffer3);
  import->set_tensor_filler(inputs[3], vec3);
}

References circlechef::as_index_vector().

---

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

• compiler/circlechef/circle/src/Op/BCQGather.h
• compiler/circlechef/circle/src/Op/BCQGather.cpp