Searched refs:averagePool (Results 1 – 1 of 1) sorted by relevance
/aosp12/packages/modules/NeuralNetworks/common/operations/ |
H A D | Pooling.cpp | 255 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool() function 404 POOLING_DISPATCH_INPUT_TYPE(averagePool, TENSOR_FLOAT32, float); in executeAveragePool() 405 POOLING_DISPATCH_INPUT_TYPE(averagePool, TENSOR_FLOAT16, _Float16); in executeAveragePool() 406 POOLING_DISPATCH_INPUT_TYPE(averagePool, TENSOR_QUANT8_ASYMM, uint8_t); in executeAveragePool() 407 POOLING_DISPATCH_INPUT_TYPE(averagePool, TENSOR_QUANT8_ASYMM_SIGNED, int8_t); in executeAveragePool()
|