jax.experimental.pallas.mosaic_gpu.Barrier# 类 jax.experimental.pallas.mosaic_gpu.Barrier(num_arrivals: 'int', num_barriers: 'int' = 1)[源代码]# 参数: num_arrivals (int) num_barriers (int) __init__(num_arrivals, num_barriers=1)# 参数: num_arrivals (int) num_barriers (int) 返回类型: None 方法 __init__(num_arrivals[, num_barriers]) get_ref_aval() 属性 num_barriers num_arrivals
