Lines Matching refs:pooling_size
35 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_qu8() local
44 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in xnnpack_average_pooling_qu8()
45 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in xnnpack_average_pooling_qu8()
61 pooling_size, pooling_size, in xnnpack_average_pooling_qu8()
114 const size_t pooling_size = state.range(3); in xnnpack_average_pooling_f32() local
123 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in xnnpack_average_pooling_f32()
124 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in xnnpack_average_pooling_f32()
140 pooling_size, pooling_size, in xnnpack_average_pooling_f32()
191 const size_t pooling_size = state.range(3); in tflite_average_pooling_f32() local
201 if (2 * padding_size == (pooling_size - 1)) { in tflite_average_pooling_f32()
210 const size_t output_height = (2 * padding_size + input_height - pooling_size) / stride + 1; in tflite_average_pooling_f32()
211 const size_t output_width = (2 * padding_size + input_width - pooling_size) / stride + 1; in tflite_average_pooling_f32()
225 pooling_size /* filter_width */, pooling_size /* filter_height */, in tflite_average_pooling_f32()