TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.experimental.auto_batching.xla

View source on GitHub

XLA utilities.

Functions

compile_nested_output(...): Wraps f with a tpu.rewrite or xla.compile, propagates output structure.