jax.experimental.pallas.tpu.BufferType# class jax.experimental.pallas.tpu.BufferType(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)[source]# Buffer type for the arguments to an emitted pipeline. __init__(*args, **kwds)# Attributes is_input is_output INPUT OUTPUT INPUT_OUTPUT MANUAL
