For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo module

pool

Structs

• ​PoolMethod:

Functions

• ​avg_pool:
• ​avg_pool_compute:
• ​avg_pool_compute_init:
• ​avg_pool_cpu: Computes the average pool.
• ​avg_pool_gpu: Computes the average pool on GPU.
• ​max_pool:
• ​max_pool_compute:
• ​max_pool_compute_init:
• ​max_pool_cpu: Computes fp32 pooling.
• ​max_pool_gpu: Computes max pooling on GPU.
• ​pool_dim_size:
• ​pool_shape:
• ​pool_shape_ceil:
• ​pool_shape_impl: Compute the output shape of a pooling operation, and assert the inputs are compatible. Works for 2D pool operations only in the NHWC format.