Module pool 

Pooling requests (max / avg, fwd + bwd) for the cuDNN actor.

Routes through CUDNN_BACKEND_OPERATION_RESAMPLE_FWD/BWD_DESCRIPTOR in the v9 frontend graph.

Structs

PoolBwdRequest

Backward pooling.

PoolFwdRequest

Forward pooling.

PoolParams

Pooling parameter struct, dim-generic.

Enums

PoolMode

Pooling op kind — kind == PoolKind::*Bwd selects the backward pass.

Functions

build_pool_bwd_graph

build_pool_fwd_graph