Lines Matching refs:inputShape
71 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
72 int32_t input_height = getSizeOfDimension(inputShape, useNchw ? 2 : 1); in initialize()
73 int32_t input_width = getSizeOfDimension(inputShape, useNchw ? 3 : 2); in initialize()
121 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
126 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
131 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
134 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in averagePoolNhwc()
138 averagePoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), in averagePoolNhwc()
144 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc() argument
149 tflite::optimized_ops::AveragePool(op_params, convertShapeToTflshape(inputShape), inputData, in averagePoolNhwc()
154 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
159 tflite::optimized_ops::L2Pool(op_params, convertShapeToTflshape(inputShape), inputData, in l2PoolNhwc()
164 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc() argument
167 std::vector<float> inputDataFloat32(getNumberOfElements(inputShape)); in l2PoolNhwc()
171 l2PoolNhwc(inputDataFloat32.data(), inputShape, param, outputDataFloat32.data(), outputShape); in l2PoolNhwc()
176 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
181 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
186 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
191 tflite::optimized_ops::MaxPool(op_params, convertShapeToTflshape(inputShape), inputData, in maxPoolNhwc()
196 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc() argument
199 std::vector<float> inputData_float32(getNumberOfElements(inputShape)); in maxPoolNhwc()
203 maxPoolNhwc(inputData_float32.data(), inputShape, param, outputData_float32.data(), in maxPoolNhwc()
210 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool() argument
214 NN_RET_CHECK(input.initialize(inputData, inputShape)); in averagePool()
223 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool() argument
227 NN_RET_CHECK(input.initialize(inputData, inputShape)); in l2Pool()
236 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool() argument
240 NN_RET_CHECK(input.initialize(inputData, inputShape)); in maxPool()