Function input_nodes

compiler/loco/src/IR/Graph.cpp:71–99

}

std::vector<Node *> input_nodes(const Graph *g)
{
  std::map<GraphInputIndex, loco::Node *> table;

  for (uint32_t n = 0; n < g->nodes()->size(); ++n)
  {
    auto node = g->nodes()->at(n);

    if (auto service = node->dialect()->service<GraphInputIndexQueryService>())
    {
      if (service->associated(node))
      {
        auto input_index = service->index(node);
        assert(table.find(input_index) == table.end());
        table[input_index] = node;
      }
    }
  }

  std::vector<loco::Node *> res;

  for (uint32_t n = 0; n < g->inputs()->size(); ++n)
  {
    auto it = table.find(n);
    res.emplace_back(it == table.end() ? nullptr : it->second);
  }

  return res;
}

std::vector<loco::Node *> output_nodes(loco::Graph *g)
{