Lines Matching refs:param

143 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param,  in averagePoolNhwc()  argument
146 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
153 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
160 averagePoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), in averagePoolNhwc()
166 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
169 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
176 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
179 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
189 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
192 auto op_params = param.toTfliteParam(outputShape); in l2PoolNhwc()
199 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
206 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
211 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
214 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
221 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
224 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
231 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
234 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
243 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
250 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
257 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool() argument
259 InputWithLayout<T> input(param.useNchw); in averagePool()
260 OutputWithLayout<T> output(param.useNchw); in averagePool()
263 NN_RET_CHECK(averagePoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in averagePool()
270 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool() argument
272 InputWithLayout<T> input(param.useNchw); in l2Pool()
273 OutputWithLayout<T> output(param.useNchw); in l2Pool()
276 NN_RET_CHECK(l2PoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in l2Pool()
283 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool() argument
285 InputWithLayout<T> input(param.useNchw); in maxPool()
286 OutputWithLayout<T> output(param.useNchw); in maxPool()
289 NN_RET_CHECK(maxPoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in maxPool()
301 PoolingParam param; in prepare() local
302 NN_RET_CHECK(param.initialize(context)); in prepare()
306 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
307 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
308 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
313 uint32_t outWidth = computeOutSize(width, param.filter_width, param.stride_width, in prepare()
314 param.padding_left, param.padding_right); in prepare()
315 uint32_t outHeight = computeOutSize(height, param.filter_height, param.stride_height, in prepare()
316 param.padding_top, param.padding_bottom); in prepare()
319 if (param.useNchw) { in prepare()
330 context->getInputShape(kInputTensor), param, \
337 PoolingParam param; in executeAveragePool() local
338 NN_RET_CHECK(param.initialize(context)); in executeAveragePool()
352 PoolingParam param; in executeL2Pool() local
353 NN_RET_CHECK(param.initialize(context)); in executeL2Pool()
365 PoolingParam param; in executeMaxPool() local
366 NN_RET_CHECK(param.initialize(context)); in executeMaxPool()