Mojo function
apply_gumbel_noise_kernel
apply_gumbel_noise_kernel[dtype: DType, input_layout: Layout, num_sms: Int, num_threads: Int](output: LayoutTensor[dtype, input_layout, MutAnyOrigin], input: LayoutTensor[dtype, input_layout, MutAnyOrigin], temperature: LegacyUnsafePointer[Float32], seed: LegacyUnsafePointer[UInt64])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!