mpqsolver::pattern::Q8SoftmaxWithQ16SubExpResolver Class Reference

#include <PatternResolver.h>

Collaboration diagram for mpqsolver::pattern::Q8SoftmaxWithQ16SubExpResolver:

Public Member Functions
std::map< luci::CircleNode *, luci::CircleQuantizer::Options::LayerParam >	resolve (const luci::Module *module) override
	resolve all nodes of Softmax pattern as prescribed
Public Member Functions inherited from mpqsolver::pattern::PatternResolver
virtual	~PatternResolver ()=default

Detailed Description

Definition at line 49 of file PatternResolver.h.

Member Function Documentation

resolve()

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

overridevirtual

resolve all nodes of Softmax pattern as prescribed

Implements mpqsolver::pattern::PatternResolver.

Definition at line 311 of file PatternResolver.cpp.

{
  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 div = dynamic_cast<luci::CircleDiv *>(node);
      if (!div)
        continue;
 
      SoftmaxPattern pattern(div);
      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;
}

References loco::active_nodes(), luci::CircleQuantizer::Options::LayerParam::name, and loco::output_nodes().

---

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

• compiler/circle-mpqsolver/src/pattern/PatternResolver.h
• compiler/circle-mpqsolver/src/pattern/PatternResolver.cpp