IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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 function

max_pool_ceil_mode_true_shape

def max_pool_ceil_mode_true_shape(input: T, filter: T, strides: T, dilations: T, paddings: T) -> IndexList[T.rank]

Computes the output shape for the mo.max_pool_ceil_mode_true graph op.

Args:

  • ​input (T): Rank-4 batched image input to the pooling operator.
  • ​filter (T): One-dimensional tensor of filter sizes on the height and width dimensions, (filter_h, filter_w).
  • ​strides (T): One-dimensional tensor of strides on the height and width dimensions, (stride_h, stride_w).
  • ​dilations (T): One-dimensional tensor of dilations on the height and width dimensions, (dilation_h, dilation_w).
  • ​paddings (T): One-dimensional tensor of paddings on the height and width dimensions, (pad_h_before, pad_h_after, pad_w_before, pad_w_after).

Returns:

IndexList[T.rank]: The output shape of the max pooling operation with ceil mode enabled.

Was this page helpful?