caffeimport::PoolingBuilder Class Reference

#include <Pooling.h>

Collaboration diagram for caffeimport::PoolingBuilder:

Public Member Functions
void	build (const ::caffe::LayerParameter &layer, GraphBuilderContext *context) const override
Public Member Functions inherited from caffeimport::GraphBuilder
virtual	~GraphBuilder ()

Detailed Description

Definition at line 27 of file Pooling.h.

Member Function Documentation

build()

void caffeimport::PoolingBuilder::build ( const ::caffe::LayerParameter &  layer, GraphBuilderContext *  context  ) const

overridevirtual

Implements caffeimport::GraphBuilder.

Definition at line 34 of file Pooling.cpp.

{
  coco::Module *module = context->module();
  coco::Block *blk = context->block();
  std::map<std::string, tensor::Shape> &shape_ctx = context->shape_ctx();
  std::map<std::string, coco::Bag *> &bag_ctx = context->bag_ctx();
 
  assert(layer.bottom().size() == 1);
  assert(layer.top().size() == 1);
 
  assert(layer.has_pooling_param());
  const auto &param = layer.pooling_param();
 
  PoolingSpec spec{param};
  {
    const auto ifm_name = layer.bottom(0);
    const auto ifm_shape = shape_ctx.at(ifm_name);
    spec.ifm_shape(ifm_shape);
  }
 
  // Create an object for an input feature map
  const auto ifm_name = layer.bottom(0);
  const auto ifm_shape = shape_ctx.at(ifm_name);
  auto ifm_bag = bag_ctx.at(ifm_name);
  auto ifm_obj = module->entity()->object()->create<coco::FeatureObject>();
 
  ifm_obj->bag(ifm_bag);
  ifm_obj->layout(coco::FeatureLayouts::BCHW::create(as_feature_shape(ifm_shape)));
 
  // Create an object for an output feature map
  const auto ofm_name = layer.top(0);
  const auto ofm_shape = spec.ofm_shape();
  auto ofm_bag = module->entity()->bag()->create(num_elements(ofm_shape));
  auto ofm_obj = module->entity()->object()->create<coco::FeatureObject>();
 
  ofm_obj->bag(ofm_bag);
  ofm_obj->layout(coco::FeatureLayouts::BCHW::create(as_feature_shape(ofm_shape)));
 
  using PoolingOpBuilder = std::function<coco::Op *(coco::Module * m, const PoolingSpec &spec)>;
 
  std::map<PoolingMethod, PoolingOpBuilder> builders;
 
  // MaxPool2D op builder
  builders[PoolingMethod::Max] = [ifm_obj](coco::Module *module, const PoolingSpec &spec) {
    auto load = op_builder(module).load(ifm_obj).pop();
 
    auto op = module->entity()->op()->create<coco::MaxPool2D>();
 
    op->arg(load);
 
    op->window()->height(spec.window_height());
    op->window()->width(spec.window_width());
 
    op->stride()->vertical(spec.vertical_stride());
    op->stride()->horizontal(spec.horizontal_stride());
 
    op->pad()->top(spec.vertical_pad());
    op->pad()->bottom(spec.vertical_pad());
    op->pad()->left(spec.horizontal_pad());
    op->pad()->right(spec.horizontal_pad());
 
    return op;
  };
 
  // AvgPool2D op builder
  builders[PoolingMethod::Avg] = [ifm_obj](coco::Module *module, const PoolingSpec &spec) {
    auto load = op_builder(module).load(ifm_obj).pop();
 
    auto op = module->entity()->op()->create<coco::AvgPool2D>();
 
    op->arg(load);
 
    // NOTE Caffe use static divisor on average pooling
    op->divisor(coco::AvgPool2D::Divisor::Static);
 
    op->window()->height(spec.window_height());
    op->window()->width(spec.window_width());
 
    op->stride()->vertical(spec.vertical_stride());
    op->stride()->horizontal(spec.horizontal_stride());
 
    op->pad()->top(spec.vertical_pad());
    op->pad()->bottom(spec.vertical_pad());
    op->pad()->left(spec.horizontal_pad());
    op->pad()->right(spec.horizontal_pad());
 
    return op;
  };
 
  // Create a pooling op
  auto builder = builders.at(spec.method());
  auto op = builder(module, spec);
 
  // Create a UnitF instruction
  auto ins = instr_builder(module).eval(ofm_obj, op);
 
  // Append the instruction to the block
  blk->instr()->append(ins);
 
  // Update bag and shape context
  bag_ctx[ofm_name] = ofm_bag;
  shape_ctx[ofm_name] = ofm_shape;
}

References coco::Op::arg(), morph::caffe::as_feature_shape(), Avg, caffeimport::GraphBuilderContext::bag_ctx(), caffeimport::GraphBuilderContext::block(), coco::FeatureLayouts::BCHW::create(), InstrBuilder::eval(), PoolingSpec::ifm_shape(), instr_builder(), OpBuilder::load(), m, Max, op_builder(), OpBuilder::pop(), caffeimport::GraphBuilderContext::shape_ctx(), and coco::AvgPool2D::Static.

---

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

• compiler/enco/frontend/caffe/src/Layer/Pooling.h
• compiler/enco/frontend/caffe/src/Layer/Pooling.cpp