Functor for 2D Average Pooling.
More...
#include <pooling.hpp>
|
| template<size_t C1, size_t C2, size_t S1, size_t S2, size_t P1, size_t P2, typename X , typename Y > |
| static void | apply (const X &x, Y &&y) |
| | Pool x into y. More...
|
| |
| template<typename X , typename Y > |
| static void | apply (const X &x, Y &&y, size_t c1, size_t c2, size_t s1, size_t s2, size_t p1, size_t p2) |
| | Pool x into y. More...
|
| |
|
|
template<typename A > |
| static constexpr bool | gpu_computable = cudnn_enabled |
| | Indicates if the temporary expression can be directly evaluated using only GPU.
|
| |
Functor for 2D Average Pooling.
◆ apply() [1/2]
template<size_t C1, size_t C2, size_t S1, size_t S2, size_t P1, size_t P2, typename X , typename Y >
| static void etl::impl::avg_pool_2d::apply |
( |
const X & |
x, |
|
|
Y && |
y |
|
) |
| |
|
inlinestatic |
Pool x into y.
- Parameters
-
| x | The expression to pol |
| y | The expression in which to store the result |
- Template Parameters
-
| C1 | The first dimension pooling ratio |
| C2 | The second dimension pooling ratio |
| S1 | The first dimension stride |
| S2 | The second dimension stride |
| P1 | The first dimension padding |
| P2 | The second dimension padding |
◆ apply() [2/2]
template<typename X , typename Y >
| static void etl::impl::avg_pool_2d::apply |
( |
const X & |
x, |
|
|
Y && |
y, |
|
|
size_t |
c1, |
|
|
size_t |
c2, |
|
|
size_t |
s1, |
|
|
size_t |
s2, |
|
|
size_t |
p1, |
|
|
size_t |
p2 |
|
) |
| |
|
inlinestatic |
Pool x into y.
- Parameters
-
| x | The expression to pol |
| y | The expression in which to store the result |
tparam C1 The first dimension pooling ratio tparam C2 The second dimension pooling ratio
tparam S1 The first dimension stride tparam S2 The second dimension stride
tparam P1 The first dimension padding tparam P2 The second dimension padding
The documentation for this struct was generated from the following file:
