circle_planner::ScratchpadHelperMCU Class Reference

#include <ScratchpadHelperMCU.h>

Collaboration diagram for circle_planner::ScratchpadHelperMCU:

Public Member Functions
uint32_t	ComputeScratchpadSizeAveragePool2d (const luci::CircleAveragePool2D *avg_pool) final
std::vector< uint32_t >	ComputeScratchpadSizeBatchMatMul (const luci::CircleBatchMatMul *batch_mat_mul) final
uint32_t	ComputeScratchpadSizeConv2d (const luci::CircleConv2D *) final
uint32_t	ComputeScratchpadSizeDepthwiseConv2d (const luci::CircleDepthwiseConv2D *depthwise_conv) final
std::vector< uint32_t >	ComputeScratchpadSizeSVDF (const luci::CircleSVDF *svdf) final
Public Member Functions inherited from circle_planner::IScratchpadHelper
virtual	~IScratchpadHelper ()=default

Detailed Description

Definition at line 25 of file ScratchpadHelperMCU.h.

Member Function Documentation

ComputeScratchpadSizeAveragePool2d()

uint32_t circle_planner::ScratchpadHelperMCU::ComputeScratchpadSizeAveragePool2d ( const luci::CircleAveragePool2D *  avg_pool )

inlinefinalvirtual

Implements circle_planner::IScratchpadHelper.

Definition at line 28 of file ScratchpadHelperMCU.h.

  {
    // for mcu AveragePool2d scratchpad tensors size = 0
    return 0;
  }

ComputeScratchpadSizeBatchMatMul()

std::vector< uint32_t > circle_planner::ScratchpadHelperMCU::ComputeScratchpadSizeBatchMatMul ( const luci::CircleBatchMatMul *  batch_mat_mul )

inlinefinalvirtual

Implements circle_planner::IScratchpadHelper.

Definition at line 35 of file ScratchpadHelperMCU.h.

  {
    throw std::runtime_error("BatchMatMul is not currently supported for mcu platform");
  }

ComputeScratchpadSizeConv2d()

uint32_t circle_planner::ScratchpadHelperMCU::ComputeScratchpadSizeConv2d ( const luci::CircleConv2D *  )

inlinefinalvirtual

Implements circle_planner::IScratchpadHelper.

Definition at line 40 of file ScratchpadHelperMCU.h.

  {
    // for mcu scratchpad size = 0
    return 0;
  }

ComputeScratchpadSizeDepthwiseConv2d()

uint32_t circle_planner::ScratchpadHelperMCU::ComputeScratchpadSizeDepthwiseConv2d ( const luci::CircleDepthwiseConv2D *  depthwise_conv )

inlinefinalvirtual

Implements circle_planner::IScratchpadHelper.

Definition at line 47 of file ScratchpadHelperMCU.h.

  {
    // for mcu DepthwiseConv2d scratchpad tensors size = 0
    return 0;
  }

ComputeScratchpadSizeSVDF()

std::vector< uint32_t > circle_planner::ScratchpadHelperMCU::ComputeScratchpadSizeSVDF ( const luci::CircleSVDF *  svdf )

inlinefinalvirtual

Implements circle_planner::IScratchpadHelper.

Definition at line 53 of file ScratchpadHelperMCU.h.

  {
    const auto svdf_input = loco::must_cast<luci::CircleNode *>(svdf->input());
    const auto weight_feature_input = loco::must_cast<luci::CircleNode *>(svdf->weight_feature());
 
    if (svdf_input->dtype() == loco::DataType::FLOAT32 and
        (weight_feature_input->dtype() == loco::DataType::S8 or
         weight_feature_input->dtype() == loco::DataType::U8))
    {
      throw std::runtime_error("Hybrid type is not currently supported for linux platform");
    }
 
    std::vector<uint32_t> scratchpad_sizes;
 
    const auto batch_size = svdf_input->dim(0).value();
    const auto num_filters = weight_feature_input->dim(0).value();
    const auto rank = svdf->svdf_rank();
    const auto num_units = num_filters / rank;
 
    if (svdf_input->dtype() == loco::DataType::S8)
    {
      scratchpad_sizes.push_back(batch_size * num_filters * sizeof(int32_t));
      scratchpad_sizes.push_back(batch_size * num_units * sizeof(int32_t));
    }
    else
    {
      scratchpad_sizes.push_back(batch_size * num_filters * sizeof(float));
    }
 
    return scratchpad_sizes;
  }

---

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

• compiler/circle-execution-plan/pal/ScratchpadHelperMCU.h