Lines Matching refs:outputShape
140 float* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
142 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
145 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
150 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
153 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in averagePoolNhwc()
157 outputShape); in averagePoolNhwc()
163 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
165 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
168 convertShapeToTflshape(outputShape), outputData); in averagePoolNhwc()
173 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc() argument
175 auto op_params = param.toTfliteParam(outputShape); in averagePoolNhwc()
180 inputData, convertShapeToTflshape(outputShape), in averagePoolNhwc()
186 float* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
188 auto op_params = param.toTfliteParam(outputShape); in l2PoolNhwc()
191 convertShapeToTflshape(outputShape), outputData); in l2PoolNhwc()
196 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc() argument
199 std::vector<float> outputDataFloat32(getNumberOfElements(outputShape)); in l2PoolNhwc()
202 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
208 float* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
210 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
213 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
218 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
220 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
223 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
228 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
230 auto op_params = param.toTfliteParam(outputShape); in maxPoolNhwc()
235 convertShapeToTflshape(outputShape), outputData); in maxPoolNhwc()
240 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc() argument
243 std::vector<float> outputData_float32(getNumberOfElements(outputShape)); in maxPoolNhwc()
247 outputShape); in maxPoolNhwc()
254 T* outputData, const Shape& outputShape) { in averagePool() argument
258 NN_RET_CHECK(output.initialize(outputData, outputShape)); in averagePool()
267 const Shape& outputShape) { in l2Pool() argument
271 NN_RET_CHECK(output.initialize(outputData, outputShape)); in l2Pool()
280 const Shape& outputShape) { in maxPool() argument
284 NN_RET_CHECK(output.initialize(outputData, outputShape)); in maxPool()