Lines Matching defs:inputShape
81 Shape inputShape = context->getInputShape(kInputTensor); in initialize() local
141 bool averagePoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
151 bool averagePoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
164 bool averagePoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
174 bool averagePoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in averagePoolNhwc()
187 bool l2PoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
197 bool l2PoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in l2PoolNhwc()
209 bool maxPoolNhwc(const float* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
219 bool maxPoolNhwc(const uint8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
229 bool maxPoolNhwc(const int8_t* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
241 bool maxPoolNhwc(const _Float16* inputData, const Shape& inputShape, const PoolingParam& param, in maxPoolNhwc()
255 bool averagePool(const T* inputData, const Shape& inputShape, const PoolingParam& param, in averagePool()
268 bool l2Pool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in l2Pool()
281 bool maxPool(const T* inputData, const Shape& inputShape, const PoolingParam& param, T* outputData, in maxPool()