Skip to main content

Mojo function

block_scales_interleave

block_scales_interleave[scales_dtype: DType, //, *, SF_VECTOR_SIZE: Int, target: StringSlice[StaticConstantOrigin] = "cpu"](output_scales_device: NDBuffer[scales_dtype, 5, MutAnyOrigin, output_scales_device.shape], input_scales_device: NDBuffer[scales_dtype, 2, MutAnyOrigin, input_scales_device.shape], ctx: DeviceContext)

Was this page helpful?