Function TEST

compiler/circle-mpqsolver/src/pattern/PatternResolver.test.cpp:136–165

} // namespace

TEST(LayerNormPatternResolverTest, resolve_pattern)
{
  auto m = luci::make_module();
  LayerNormTestGraph g;
  g.init();
  g.transfer_to(m.get());

  std::map<luci::CircleNode *, LayerParam> params;
  mpqsolver::pattern::Q8LayerNormWithQ16VarianceResolver resolver;
  EXPECT_NO_THROW({ params = resolver.resolve(m.get()); });

  std::set<luci::CircleNode *> q16_nodes = {g.sub_squared, g.mean_as_variance, g.add_eps, g.rsqrt};
  std::set<luci::CircleNode *> q8_nodes = {g.mean_of_ifm, g.sub, g.mul};

  // params of all valid layers are set
  EXPECT_EQ(params.size(), q16_nodes.size() + q8_nodes.size());

  for (auto param : params)
  {
    // params of all layers are set as prescribed
    if (q16_nodes.find(param.first) != q16_nodes.end())
    {
      EXPECT_STREQ(param.second.dtype.c_str(), "int16");
    }
    else if (q8_nodes.find(param.first) != q8_nodes.end())
    {
      EXPECT_STREQ(param.second.dtype.c_str(), "uint8");
    }
  }
}

TEST(LayerNormPatternResolverTest, resolve_pattern_NEG)
{