ONE - On-device Neural Engine
|
#include <backend/IPortableTensor.h>
#include <ir/DataType.h>
#include <ir/Layout.h>
#include <libnpuhost.h>
#include <type_traits>
Go to the source code of this file.
Namespaces | |
namespace | onert |
namespace | onert::backend |
namespace | onert::backend::trix |
Functions | |
data_type | onert::backend::trix::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 | onert::backend::trix::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 | onert::backend::trix::setBuffers (const std::vector< T * > &tensors, generic_buffers *buf) |
Set the generic_buffers object. | |