nnfw_api_wrapper_pybind.cc File Reference

#include <pybind11/pybind11.h>
#include "nnfw_session_bindings.h"
#include "nnfw_tensorinfo_bindings.h"
#include "nnfw_traininfo_bindings.h"

Go to the source code of this file.

Functions
	PYBIND11_MODULE (libnnfw_api_pybind, m)

Function Documentation

PYBIND11_MODULE()

PYBIND11_MODULE	(	libnnfw_api_pybind	,
		m
	)

Definition at line 25 of file nnfw_api_wrapper_pybind.cc.

{
  m.doc() = "Main module that contains infer and experimental submodules";
 
  // Bind common `NNFW_SESSION` class
  bind_nnfw_session(m);
 
  // Bind `NNFW_SESSION` class for inference
  // Currently, the `infer` session is the same as common.
  auto infer = m.def_submodule("infer", "Inference submodule");
  infer.attr("nnfw_session") = m.attr("nnfw_session");
 
  // Bind experimental `NNFW_SESSION` class
  auto experimental = m.def_submodule("experimental", "Experimental submodule");
  experimental.attr("nnfw_session") = m.attr("nnfw_session");
  bind_experimental_nnfw_session(experimental);
 
  // Bind common `tensorinfo` class
  bind_tensorinfo(m);
 
  m.doc() = "NNFW Python Bindings for Training";
 
  // Bind training enums
  bind_nnfw_train_enums(m);
 
  // Bind training nnfw_loss_info
  bind_nnfw_loss_info(m);
 
  // Bind_train_info
  bind_nnfw_train_info(m);
}

References onert::api::python::bind_experimental_nnfw_session(), bind_nnfw_loss_info(), onert::api::python::bind_nnfw_session(), bind_nnfw_train_enums(), bind_nnfw_train_info(), onert::api::python::bind_tensorinfo(), and m.