Lines Matching refs:pooling_elements

60   inline ArgMaxPoolMicrokernelTester& pooling_elements(size_t pooling_elements) {  in pooling_elements()  function
61     assert(pooling_elements != 0);  in pooling_elements()
62     this->pooling_elements_ = pooling_elements;  in pooling_elements()
66   inline size_t pooling_elements() const {  in pooling_elements()  function
71     if (pooling_elements() <= primary_pooling_tile()) {  in packed_pooling_elements()
74 …return (pooling_elements() - primary_pooling_tile()) % incremental_pooling_tile() == 0 ? pooling_e…  in packed_pooling_elements()
153       ((output_pixels() - 1) * step() + pooling_elements()) * channels());
162       for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
166         indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
173           for (size_t p = 0; p < pooling_elements(); p++) {
186       argmaxpool(output_pixels(), pooling_elements(), channels(),
196             << ", pooling elements = " << pooling_elements() << ", step = " << step()
202             << ", pooling elements = " << pooling_elements() << ", step = " << step()
206             << ", pooling elements = " << pooling_elements() << ", step = " << step()
220       ((output_pixels() - 1) * step() + pooling_elements()) * channels());
233       for (size_t i = 0; i < (output_pixels() - 1) * step() + pooling_elements(); i++) {
237         indirect_input.begin() + (output_pixels() - 1) * step() + pooling_elements(), rng);
244           for (size_t p = 0; p < pooling_elements(); p++) {
257       argmaxpool(output_pixels(), pooling_elements(), channels(),
269             << ", pooling elements = " << pooling_elements() << ", step = " << step()
275             << ", pooling elements = " << pooling_elements() << ", step = " << step()
279             << ", pooling elements = " << pooling_elements() << ", step = " << step()