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.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit