Lines Matching refs:param
139 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
142 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
149 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
156 averagePoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), in averagePoolNhwc()
162 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
165 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
172 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
175 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
185 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
188 auto op_params = param.toTfliteParam(outputShape); in l2PoolNhwc()
195 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
202 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
207 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
210 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
217 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
220 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
227 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
230 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
239 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
246 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
253 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool() argument
255 InputWithLayout<T> input(param.useNchw); in averagePool()
256 OutputWithLayout<T> output(param.useNchw); in averagePool()
259 NN_RET_CHECK(averagePoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in averagePool()
266 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool() argument
268 InputWithLayout<T> input(param.useNchw); in l2Pool()
269 OutputWithLayout<T> output(param.useNchw); in l2Pool()
272 NN_RET_CHECK(l2PoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in l2Pool()
279 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool() argument
281 InputWithLayout<T> input(param.useNchw); in maxPool()
282 OutputWithLayout<T> output(param.useNchw); in maxPool()
285 NN_RET_CHECK(maxPoolNhwc(input.getNhwcBuffer(), input.getNhwcShape(), param, in maxPool()
360 PoolingParam param; in prepare() local
361 NN_RET_CHECK(param.initialize(context)); in prepare()
365 uint32_t height = getSizeOfDimension(input, param.useNchw ? 2 : 1); in prepare()
366 uint32_t width = getSizeOfDimension(input, param.useNchw ? 3 : 2); in prepare()
367 uint32_t channels = getSizeOfDimension(input, param.useNchw ? 1 : 3); in prepare()
372 uint32_t outWidth = computeOutSize(width, param.filter_width, param.stride_width, in prepare()
373 param.padding_left, param.padding_right); in prepare()
374 uint32_t outHeight = computeOutSize(height, param.filter_height, param.stride_height, in prepare()
375 param.padding_top, param.padding_bottom); in prepare()
378 if (param.useNchw) { in prepare()
389 context->getInputShape(kInputTensor), param, \
396 PoolingParam param; in executeAveragePool() local
397 NN_RET_CHECK(param.initialize(context)); in executeAveragePool()
411 PoolingParam param; in executeL2Pool() local
412 NN_RET_CHECK(param.initialize(context)); in executeL2Pool()
424 PoolingParam param; in executeMaxPool() local
425 NN_RET_CHECK(param.initialize(context)); in executeMaxPool()