ONE - On-device Neural Engine
All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Modules Pages
nnfw_api_wrapper.h File Reference
#include "nnfw.h"
#include "nnfw_experimental.h"
#include <pybind11/stl.h>
#include <pybind11/numpy.h>

Go to the source code of this file.

Data Structures

struct  onert::api::python::tensorinfo
 tensor info describes the type and shape of tensors More...
 
class  onert::api::python::NNFW_SESSION
 

Namespaces

namespace  onert
 
namespace  onert::api
 
namespace  onert::api::python
 

Functions

void onert::api::python::ensure_status (NNFW_STATUS status)
 Handle errors with NNFW_STATUS in API functions.
 
NNFW_LAYOUT onert::api::python::getLayout (const char *layout="")
 
NNFW_TYPE onert::api::python::getType (const char *type="")
 
const char * onert::api::python::getStringType (NNFW_TYPE type)
 
uint64_t onert::api::python::num_elems (const nnfw_tensorinfo *tensor_info)
 Get the total number of elements in nnfw_tensorinfo->dims.
 
py::list onert::api::python::get_dims (const tensorinfo &tensor_info)
 Get nnfw_tensorinfo->dims.
 
void onert::api::python::set_dims (tensorinfo &tensor_info, const py::list &array)
 Set nnfw_tensorinfo->dims.