Skip to main content

Module repeat

Module repeat 

Source
Expand description

repeat plan — per-axis tile (output > input). PyTorch torch.repeat(x, *repeats): output.shape[d] = input.shape[d] * repeats[d]. The kernel walks output cells and computes input coords as output_coord[d] % input.shape[d].

Structs§

RepeatArgs
Args bundle for a Repeat launch.
RepeatDescriptor
Descriptor for a repeat op.
RepeatPlan
repeat plan.