Expression Templates Library (ETL)
Static Public Member Functions | List of all members
etl::impl::cudnn::max_pool_upsample_3d Struct Reference

Functor for 3D Max Pooling Upsample. More...

#include <pooling_upsample.hpp>

Static Public Member Functions

template<typename A , typename B , typename C , typename M >
static void apply ([[maybe_unused]] A &&in, [[maybe_unused]] B &&out, [[maybe_unused]] C &&errors, [[maybe_unused]] M &m, [[maybe_unused]] size_t c1, [[maybe_unused]] size_t c2, [[maybe_unused]] size_t c3)
 Apply the functor on sub and store the result in m. More...
 

Detailed Description

Functor for 3D Max Pooling Upsample.

Member Function Documentation

◆ apply()

template<typename A , typename B , typename C , typename M >
static void etl::impl::cudnn::max_pool_upsample_3d::apply ( [[maybe_unused] ] A &&  in,
[[maybe_unused] ] B &&  out,
[[maybe_unused] ] C &&  errors,
[[maybe_unused] ] M &  m,
[[maybe_unused] ] size_t  c1,
[[maybe_unused] ] size_t  c2,
[[maybe_unused] ] size_t  c3 
)
inlinestatic

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

Parameters
inThe sub expression
mThe storage matrix
c1The first dimension pooling ratio
c2The second dimension pooling ratio

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