ONE - On-device Neural Engine
Loading...
Searching...
No Matches
TensorUtils.h File Reference
#include "cker/Types.h"
#include "cker/PortableTensorUtils.h"
#include "cker/NeonTensorUtils.h"
#include "cker/neon/neon_check.h"
#include <cstring>
#include <cmath>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

void nnfw::cker::CwiseClipping (float *vector, const int v_size, const float clipping_value)
 
void nnfw::cker::VectorBatchVectorAdd (const float *vector, int v_size, int n_batch, float *batch_vector)
 
void nnfw::cker::VectorBatchVectorAssign (const float *vector, int v_size, int n_batch, float *batch_vector)
 
template<typename T >
void nnfw::cker::VectorVectorCwiseProduct (const T *__restrict__ vector1, const T *__restrict__ vector2, int v_size, T *__restrict__ result)
 
template<typename T >
void nnfw::cker::VectorVectorCwiseProductAccumulate (const T *__restrict__ vector1, const T *__restrict__ vector2, int v_size, T *__restrict__ result)
 
template<typename T >
void nnfw::cker::VectorBatchVectorCwiseProduct (const T *vector, int v_size, const T *batch_vector, int n_batch, T *result)
 
template<typename T >
void nnfw::cker::VectorBatchVectorCwiseProductAccumulate (const T *vector, int v_size, const T *batch_vector, int n_batch, T *result)
 
bool nnfw::cker::IsZeroVector (const float *vector, int v_size)
 
void nnfw::cker::ApplyActivationToVector (const float *vector, int v_size, FusedActivationFunctionType activation, float *result)
 
void nnfw::cker::Sub1Vector (const float *vector, int v_size, float *result)
 
void nnfw::cker::SymmetricQuantizeFloats (const float *values, const int size, int8_t *quantized_values, float *min, float *max, float *scaling_factor)
 
void nnfw::cker::MatrixBatchVectorMultiplyAccumulate (const int8_t *matrix, const int m_rows, const int m_cols, const int8_t *vector, const float *scaling_factors, int n_batch, float *result, int result_stride)
 
void nnfw::cker::MatrixBatchVectorMultiplyAccumulate (const float *matrix, int m_rows, int m_cols, const float *vector, int n_batch, float *result, int result_stride)
 
void nnfw::cker::MatrixBatchVectorMultiplyAccumulate (const int8_t *matrix, const int m_rows, const int m_cols, const int8_t *vectors, const float *scaling_factors, int n_batch, int32_t *scratch, float *result, int result_stride, ruy::Context *ruy_context)
 
void nnfw::cker::MeanStddevNormalization (const float *input_vector, float *output_vector, int v_size, int n_batch)
 
void nnfw::cker::ZeroVector (float *vector, int v_size)