ONE - On-device Neural Engine
|
tensor info describes the type and shape of tensors More...
#include <nnfw_api_wrapper.h>
Data Fields | |
const char * | dtype |
int32_t | rank |
int32_t | dims [NNFW_MAX_RANK] |
tensor info describes the type and shape of tensors
This structure is used to describe input and output tensors. Application can get input and output tensor type and shape described in model by using input_tensorinfo
and output_tensorinfo
Maximum rank is 6 (NNFW_MAX_RANK). And tensor's dimension value is filled in 'dims' field from index 0. For example, if tensor's rank is 4, application can get dimension value from dims[0], dims[1], dims[2], and dims[3]
Definition at line 47 of file nnfw_api_wrapper.h.
int32_t onert::api::python::tensorinfo::dims[NNFW_MAX_RANK] |
The dimension of tensor. Maximum rank is 6 (NNFW_MAX_RANK).
Definition at line 57 of file nnfw_api_wrapper.h.
Referenced by onert::api::python::get_dims(), onert::api::python::NNFW_SESSION::input_tensorinfo(), onert::api::python::NNFW_SESSION::output_tensorinfo(), onert::api::python::set_dims(), and onert::api::python::NNFW_SESSION::set_input_tensorinfo().
const char* onert::api::python::tensorinfo::dtype |
The data type
Definition at line 50 of file nnfw_api_wrapper.h.
Referenced by onert::api::python::bind_tensorinfo(), onert::api::python::NNFW_SESSION::input_tensorinfo(), onert::api::python::NNFW_SESSION::output_tensorinfo(), and onert::api::python::NNFW_SESSION::set_input_tensorinfo().
int32_t onert::api::python::tensorinfo::rank |
The number of dimensions (rank)
Definition at line 52 of file nnfw_api_wrapper.h.
Referenced by onert::api::python::bind_tensorinfo(), onert::api::python::get_dims(), onert::api::python::NNFW_SESSION::input_tensorinfo(), onert::api::python::NNFW_SESSION::output_tensorinfo(), onert::api::python::set_dims(), and onert::api::python::NNFW_SESSION::set_input_tensorinfo().