jax.experimental.pallas.mosaic_gpu.SMEM

jax.experimental.pallas.mosaic_gpu.SMEM = GPUMemorySpace.SMEM

別名為 jax.experimental.pallas.mosaic_gpu.GPUMemorySpace.SMEM。

參數:

• shape (tuple[int, ...])

• dtype (jnp.dtype)

• transforms (Sequence[MemoryRefTransform])

回傳型別:

pallas_core.MemoryRef