etl::impl::cudnn::avg_pool_3d Struct Reference

Functor for 2D Max Pooling. More...

#include <max_pooling.hpp>

Static Public Member Functions
template<typename X , typename Y >
static void	apply ([[maybe_unused]] const X &x, [[maybe_unused]] Y &&y, [[maybe_unused]] size_t c1, [[maybe_unused]] size_t c2, [[maybe_unused]] size_t c3, [[maybe_unused]] size_t s1, [[maybe_unused]] size_t s2, [[maybe_unused]] size_t s3, [[maybe_unused]] size_t p1, [[maybe_unused]] size_t p2, [[maybe_unused]] size_t p3)
	Apply the functor on sub and store the result in m. More...

Detailed Description

Functor for 2D Max Pooling.

Member Function Documentation

apply()

template<typename X , typename Y >

static void etl::impl::cudnn::avg_pool_3d::apply ( [[maybe_unused] ] const X &  x, [[maybe_unused] ] Y &&  y, [[maybe_unused] ] size_t  c1, [[maybe_unused] ] size_t  c2, [[maybe_unused] ] size_t  c3, [[maybe_unused] ] size_t  s1, [[maybe_unused] ] size_t  s2, [[maybe_unused] ] size_t  s3, [[maybe_unused] ] size_t  p1, [[maybe_unused] ] size_t  p2, [[maybe_unused] ] size_t  p3  )

inlinestatic

Apply the functor on sub and store the result in m.

Parameters

x	The sub expression
y	The storage matrix
c1	The first dimension pooling ratio
c2	The second dimension pooling ratio

---

The documentation for this struct was generated from the following file:

• include/etl/impl/cudnn/max_pooling.hpp