ONE - On-device Neural Engine
Loading...
Searching...
No Matches
RuySupport.h File Reference
#include <util/ConfigSource.h>
#include <ruy/matrix.h>
#include <ruy/ruy.h>
#include <cassert>
#include "Types.h"

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::ruy
 
namespace  nnfw::ruy::ruy_support
 

Functions

inline ::ruy::CachePolicy nnfw::ruy::ruy_support::ToRuyCachePolicy (CachePolicy cache_policy)
 
template<typename Scalar , typename DataPointer >
void nnfw::ruy::ruy_support::MakeRuyMatrix (const MatrixParams< Scalar > &params, DataPointer data_ptr, ::ruy::Matrix< Scalar > *dst, bool use_caching=false)
 
template<typename AccumScalar , typename DstScalar , QuantizationFlavor quantization_flavor>
void nnfw::ruy::ruy_support::MakeRuyMulParams (const GemmParams< AccumScalar, DstScalar, quantization_flavor > &params, ::ruy::MulParams< AccumScalar, DstScalar > *ruy_mul_params)
 
template<typename DstScalar , QuantizationFlavor quantization_flavor>
void nnfw::ruy::ruy_support::MakeRuyMulParams (const GemmParams< std::int32_t, DstScalar, quantization_flavor > &params, ::ruy::MulParams< std::int32_t, DstScalar > *ruy_mul_params)
 
template<QuantizationFlavor quantization_flavor>
void nnfw::ruy::ruy_support::MakeRuyMulParams (const GemmParams< std::int32_t, std::int32_t, quantization_flavor > &params, ::ruy::MulParams< std::int32_t, std::int32_t > *ruy_mul_params)