ONE - On-device Neural Engine
|
Namespaces | |
namespace | ops |
Data Structures | |
class | Backend |
class | BackendContext |
class | BatchThreadPool |
Class that has a threadpool for batch-by-batch multi-threading. More... | |
class | Config |
class | DevContext |
NPU device context of trix backend. More... | |
class | KernelGenerator |
Typedefs | |
using | ModelID = uint32_t |
using | Tensor = basic::Tensor |
using | ExternalTensor = basic::ExternalTensor |
using | TensorBuilder = basic::TensorBuilder |
Functions | |
data_type | convertDataType (const ir::DataType type) |
Convert type of data from onert type to npu type. | |
template<typename T , std::enable_if_t< std::is_base_of< IPortableTensor, T >::value, bool > = true> | |
void | setDataInfo (const std::vector< T * > &tensors, tensors_data_info *info) |
Set the tensors_data_info object. | |
template<typename T , std::enable_if_t< std::is_base_of< IPortableTensor, T >::value, bool > = true> | |
void | setBuffers (const std::vector< T * > &tensors, generic_buffers *buf) |
Set the generic_buffers object. | |
using onert::backend::trix::ExternalTensor = typedef basic::ExternalTensor |
using onert::backend::trix::ModelID = typedef uint32_t |
Definition at line 34 of file DevContext.h.
using onert::backend::trix::Tensor = typedef basic::Tensor |
using onert::backend::trix::TensorBuilder = typedef basic::TensorBuilder |
Definition at line 29 of file TensorBuilder.h.
data_type onert::backend::trix::convertDataType | ( | const ir::DataType | type | ) |
Convert type of data from onert type to npu type.
type | Data type in onert |
Definition at line 26 of file Convert.cc.
Referenced by setDataInfo().
void onert::backend::trix::setBuffers | ( | const std::vector< T * > & | tensors, |
generic_buffers * | buf | ||
) |
Set the generic_buffers object.
T | Type of tensor based of IPortableTensor |
tensors | Tensors that have buffer information |
buf | generic_buffers to be set |
Definition at line 69 of file Convert.h.
Referenced by onert::backend::trix::ops::BulkLayer::run().
void onert::backend::trix::setDataInfo | ( | const std::vector< T * > & | tensors, |
tensors_data_info * | info | ||
) |
Set the tensors_data_info object.
T | Type of tensor based of IPortableTensor |
tensors | Tensors that have data information |
info | tensors_data_info to be set |
Definition at line 50 of file Convert.h.
References convertDataType(), and info.
Referenced by onert::backend::trix::ops::BulkLayer::run().