jax.experimental.pallas.mosaic_gpu.SwizzleTransform# class jax.experimental.pallas.mosaic_gpu.SwizzleTransform(swizzle: 'int')[source]# Parameters: swizzle (int) __init__(swizzle)# Parameters: swizzle (int) Return type: None Methods __init__(swizzle) pretty_print(context) transform_type(x) undo(x) Attributes swizzle
