• Home
  • Raw
  • Download

Lines Matching refs:Shape

81             Shape inputShape = context->getInputShape(kInputTensor);  in initialize()
105 tflite::PoolParams toTfliteParam(const Shape& output) const { in toTfliteParam()
141 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
142 float* outputData, const Shape& outputShape) { in averagePoolNhwc()
151 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
152 _Float16* outputData, const Shape& outputShape) { in averagePoolNhwc()
164 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
165 uint8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
174 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
175 int8_t* outputData, const Shape& outputShape) { in averagePoolNhwc()
187 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
188 float* outputData, const Shape& outputShape) { in l2PoolNhwc()
197 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
198 _Float16* outputData, const Shape& outputShape) { in l2PoolNhwc()
209 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
210 float* outputData, const Shape& outputShape) { in maxPoolNhwc()
219 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
220 uint8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
229 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
230 int8_t* outputData, const Shape& outputShape) { in maxPoolNhwc()
241 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
242 _Float16* outputData, const Shape& outputShape) { in maxPoolNhwc()
255 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool()
256 T* outputData, const Shape& outputShape) { in averagePool()
268 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool()
269 const Shape& outputShape) { in l2Pool()
281 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool()
282 const Shape& outputShape) { in maxPool()
362 Shape input = context->getInputShape(kInputTensor); in prepare()
382 Shape output = input; in prepare()