triton.experimental.gluon.nvidia.hopper.TensorDescriptorIm2Col

class triton.experimental.gluon.nvidia.hopper.TensorDescriptorIm2Col(base: Any, shape: List[int], strides: List[int], block_shape: List[int], layout: triton.experimental.gluon.language._layouts.NVMMASharedLayout, padding: str = 'zero', round_f32_to_tf32: bool = False, element_strides: List[int] | None = None, pixel_box_lower_corner: List[int] | None = None, pixel_box_upper_corner: List[int] | None = None)

__init__(self, base: Any, shape: List[int], strides: List[int], block_shape: List[int], layout: NVMMASharedLayout, padding: str = 'zero', round_f32_to_tf32: bool = False, element_strides: List[int] | None = None, pixel_box_lower_corner: List[int] | None = None, pixel_box_upper_corner: List[int] | None = None) → None

Methods

__init__(self, base, shape, strides, ...[, ...])
from_tensor(tensor, block_shape, layout[, ...])	Create a TensorDescriptorIm2Col from a tensor.

Attributes

element_strides
mode
padding
pixel_box_lower_corner
pixel_box_upper_corner
round_f32_to_tf32
base
shape
strides
block_shape
layout