Lines Matching refs:outputData
140 float* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
145 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
150 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
158 convertFloat32ToFloat16(outputDataFloat32, outputData); in averagePoolNhwc()
163 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
168 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
173 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
181 outputData); in averagePoolNhwc()
186 float* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
191 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc()
196 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
203 convertFloat32ToFloat16(outputDataFloat32, outputData); in l2PoolNhwc()
208 float* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
213 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
218 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
223 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
228 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
235 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
240 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
248 convertFloat32ToFloat16(outputData_float32, outputData); in maxPoolNhwc()
254 T* outputData, const Shape& outputShape) { in averagePool() argument
258 NN_RET_CHECK(output.initialize(outputData, outputShape)); in averagePool()
266 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool() argument
271 NN_RET_CHECK(output.initialize(outputData, outputShape)); in l2Pool()
279 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool() argument
284 NN_RET_CHECK(output.initialize(outputData, outputShape)); in maxPool()