tfp.experimental.auto_batching.TensorType

TensorType(dtype, shape)

dtype

shape