ONE - On-device Neural Engine
Loading...
Searching...
No Matches
AveragePool.h File Reference
#include "cker/neon/neon_check.h"
#include "cker/eigen/Utils.h"
#include "cker/Shape.h"
#include "cker/Types.h"
#include "cker/Utils.h"
#include <Eigen/Core>

Go to the source code of this file.

Namespaces

namespace  nnfw
 
namespace  nnfw::cker
 

Functions

template<typename T >
void nnfw::cker::AveragePool (const PoolParams &, const Shape &, const T *, const Shape &, T *)
 
template<>
void nnfw::cker::AveragePool< float > (const PoolParams &params, const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
 
void nnfw::cker::AveragePool16 (const PoolParams &params, const Shape &input_shape, const uint8_t *input_data, const Shape &output_shape, uint8_t *output_data)
 
void nnfw::cker::AveragePool32 (const PoolParams &params, const Shape &input_shape, const uint8_t *input_data, const Shape &output_shape, uint8_t *output_data)
 
template<>
void nnfw::cker::AveragePool< uint8_t > (const PoolParams &params, const Shape &input_shape, const uint8_t *input_data, const Shape &output_shape, uint8_t *output_data)
 
template<>
void nnfw::cker::AveragePool< int8_t > (const PoolParams &params, const Shape &input_shape, const int8_t *input_data, const Shape &output_shape, int8_t *output_data)