Lines Matching refs:inputShape
81 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
82 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
83 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
141 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
146 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
151 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
154 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in averagePoolNhwc()
158 averagePoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), in averagePoolNhwc()
164 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
169 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
174 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
181 tflite::reference_integer_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), in averagePoolNhwc()
187 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
192 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc()
197 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
200 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in l2PoolNhwc()
204 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
209 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
214 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
219 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
224 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
229 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
236 tflite::reference_integer_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
241 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
244 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in maxPoolNhwc()
248 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
255 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool() argument
259 NN_RET_CHECK(input.initialize(inputData, inputShape)); in averagePool()
268 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool() argument
272 NN_RET_CHECK(input.initialize(inputData, inputShape)); in l2Pool()
281 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool() argument
285 NN_RET_CHECK(input.initialize(inputData, inputShape)); in maxPool()