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 struct
TPoolPatchMerger
struct TPoolPatchMerger
Registers the tpool_patch_merger graph op with the graph compiler.
Implemented traits
Methods
execute
static def execute[dtype: DType, //, target: StringSlice[ImmStaticOrigin]](output: ManagedTensorSlice[IOSpec[_, _].Output, static_spec=output.static_spec], input: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=input.static_spec], grid_thws: ManagedTensorSlice[IOSpec[_, _].Input, static_spec=grid_thws.static_spec], kH: Int32, kW: Int32, max_h: Int32, max_w: Int32, _total_output_patches: Int32, ctx: DeviceContext)