tfp.experimental.substrates.jax.util.DeferredTensor

View source on GitHub