nnc::InterpreterBackend Class Reference

#include <InterpreterBackend.h>

Public Member Functions
	InterpreterBackend (std::string input_dir, std::string output_dir)
void	run (mir::Graph *data)

Detailed Description

Definition at line 27 of file InterpreterBackend.h.

Constructor & Destructor Documentation

InterpreterBackend()

nnc::InterpreterBackend::InterpreterBackend	(	std::string	input_dir,
		std::string	output_dir
	)

Definition at line 132 of file InterpreterBackend.cpp.

  : _input_dir(std::move(input_dir)), _output_dir(std::move(output_dir))
{
}

Member Function Documentation

run()

void nnc::InterpreterBackend::run

(

mir::Graph *

data

)

Definition at line 137 of file InterpreterBackend.cpp.

{
  assert(graph);
 
  mir_interpreter::MIRInterpreter interpreter;
 
  for (const auto *input_op : graph->getInputs())
  {
    const Operation::Output *input = input_op->getOutput(0);
 
    std::string tensor_name = input->getName();
    assert(!tensor_name.empty());
    std::replace(tensor_name.begin(), tensor_name.end(), '/', '_');
    std::string filename = _input_dir + "/" + tensor_name + ".dat";
 
    TensorVariant tensor = readTensorFromFile(filename, input->getType());
    interpreter.setTensor(input, std::move(tensor));
  }
 
  graph->accept(&interpreter);
 
  for (const auto *output_op : graph->getOutputs())
  {
    const auto &output_name = output_op->getInput(0)->getName();
 
#ifdef NNC_HDF5_SUPPORTED
    const auto &tensor = interpreter.getTensor(output_op->getInput(0));
    writeTensorToHDF5File(tensor, output_name, _output_dir);
#else
    std::cout << "Result <" << output_name << "> wasn't saved, due to lack of HDF5" << std::endl;
#endif // NNC_HDF5_SUPPORTED
  }
}

Referenced by package.infer.session::inference().

---

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

• compiler/nnc/include/backends/interpreter/InterpreterBackend.h
• compiler/nnc/backends/interpreter/InterpreterBackend.cpp