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 "cker/Types.h"

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 
namespace  nnfw::cker::ruy_support
 

Functions

ruy::CachePolicy nnfw::cker::ruy_support::ToRuyCachePolicy (CachePolicy cache_policy)
 
template<typename Scalar , typename DataPointer >
void nnfw::cker::ruy_support::MakeRuyMatrix (const MatrixParams< Scalar > &params, DataPointer data_ptr, ruy::Matrix< Scalar > *dst, bool use_caching=false)
 
template<typename DstScalar , QuantizationFlavor quantization_flavor>
void nnfw::cker::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::cker::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)