Function interpret

compiler/circle-eval-diff/src/CircleEvalDiff.cpp:109–140

{

std::vector<std::shared_ptr<Tensor>> interpret(const luci::Module *module,
                                               const InputDataLoader::Data &data)
{
  auto interpreter = std::make_unique<luci_interpreter::Interpreter>(module);

  auto input_nodes = ::inputs_of(module);
  auto output_nodes = ::outputs_of(module);

  for (uint32_t input_idx = 0; input_idx < data.size(); input_idx++)
  {
    auto input_node = loco::must_cast<const luci::CircleInput *>(input_nodes[input_idx]);
    assert(input_node->index() == input_idx);

    auto input_data = data.at(input_idx);
    interpreter->writeInputTensor(input_node, input_data.buffer(), input_data.byte_size());
  }

  interpreter->interpret();

  std::vector<std::shared_ptr<Tensor>> outputs;
  for (uint32_t output_idx = 0; output_idx < output_nodes.size(); output_idx++)
  {
    auto output_node = loco::must_cast<const luci::CircleOutput *>(output_nodes[output_idx]);
    assert(output_node->index() == output_idx);

    auto tensor = createEmptyTensor(output_node);
    interpreter->readOutputTensor(output_node, tensor->buffer(), tensor->byte_size());
    outputs.emplace_back(tensor);
  }

  return outputs;
}

CircleEvalDiff::CircleEvalDiff(std::unique_ptr<Context> &&ctx) : _ctx(std::move(ctx))
{