Lines Matching refs:outputShape
142 float* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
144 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
147 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
152 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
155 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in averagePoolNhwc()
159 outputShape); in averagePoolNhwc()
165 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
167 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
170 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
175 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
177 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
182 inputData, convertShapeToTflshape(outputShape), in averagePoolNhwc()
188 float* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
190 auto op_params = param.toTfliteParam(outputShape); in l2PoolNhwc()
193 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc()
198 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
201 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in l2PoolNhwc()
204 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
210 float* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
212 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
215 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
220 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
222 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
225 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
230 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
232 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
237 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
242 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
245 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in maxPoolNhwc()
249 outputShape); in maxPoolNhwc()
256 T* outputData, const Shape& outputShape) { in averagePool() argument
260 NN_RET_CHECK(output.initialize(outputData, outputShape)); in averagePool()
269 const Shape& outputShape) { in l2Pool() argument
273 NN_RET_CHECK(output.initialize(outputData, outputShape)); in l2Pool()
282 const Shape& outputShape) { in maxPool() argument
286 NN_RET_CHECK(output.initialize(outputData, outputShape)); in maxPool()