circlechef::CircleOpBCQFullyConnected Class Reference

circlechef operator builder for BCQFullyConnected More...

#include <BCQFullyConnected.h>

Collaboration diagram for circlechef::CircleOpBCQFullyConnected:

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 BCQFullyConnected

Definition at line 28 of file BCQFullyConnected.h.

Member Function Documentation

build()

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

overridevirtual

Implements circlechef::CircleOpChef.

Definition at line 45 of file BCQFullyConnected.cpp.

{
  auto op_params = op->builtin_options_as_BCQFullyConnectedOptions();
  assert(op_params != nullptr);
 
  auto operation = model_recipe->add_operation();
 
  operation->set_type("BCQFullyConnected");
 
  auto op_options = operation->mutable_bcq_fully_connected_options();
 
  op_options->set_weights_hidden_size(op_params->weights_hidden_size());
  op_options->set_activation(as_circlechef_activation(op_params->fused_activation_function()));
 
  return operation;
}

References circlechef::as_circlechef_activation().

filler()

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

overridevirtual

Implements circlechef::CircleOpChef.

Definition at line 24 of file BCQFullyConnected.cpp.

{
  const std::vector<int32_t> &inputs = as_index_vector(op->inputs());
 
  import->set_tensor_filler(inputs[1]);
  import->set_tensor_filler(inputs[3]);
 
  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 *tensor4 = import->tensors()->Get(inputs[4]);
  assert(tensor4->type() == circle::TensorType::TensorType_INT32);
  const circle::Buffer *buffer4 = import->buffers()->Get(tensor4->buffer());
  auto vec4 = extract_buffer<int32_t>(buffer4);
  import->set_tensor_filler(inputs[4], vec4);
}

References circlechef::as_index_vector().

---

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

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