Method resolve

compiler/circle-mpqsolver/src/pattern/PatternResolver.cpp:269–308

} // namespace

std::map<luci::CircleNode *, LayerParam>
Q8LayerNormWithQ16VarianceResolver::resolve(const luci::Module *module)
{
  if (!module)
  {
    throw std::runtime_error("No module for pattern resolving");
  }

  std::map<luci::CircleNode *, LayerParam> nodes_params;
  for (size_t idx = 0; idx < module->size(); ++idx)
  {
    auto graph = module->graph(idx);

    for (auto node : loco::active_nodes(loco::output_nodes(graph)))
    {
      auto const mul = dynamic_cast<luci::CircleMul *>(node);
      if (!mul)
        continue;

      LayerNormPattern pattern(mul);
      if (!pattern.matched())
        continue;

      // set quantization parameters of recognized pattern
      for (auto q16_node : pattern.get_q16_nodes())
      {
        LayerParam param = {q16_node->name(), "int16", "channel"};
        nodes_params[q16_node] = param;
      }

      for (auto q8_node : pattern.get_q8_nodes())
      {
        LayerParam param = {q8_node->name(), "uint8", "channel"};
        nodes_params[q8_node] = param;
      }
    }
  }

  return nodes_params;
}

std::map<luci::CircleNode *, LayerParam>
Q8SoftmaxWithQ16SubExpResolver::resolve(const luci::Module *module)