ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfp.experimental.auto_batching.tf_backend

TensorFlow (graph) backend for auto-batching VM.

Implements VM variable stack and registers backed by TF Tensors.

Classes

class TensorFlowBackend: Implements the TF backend ops for a PC auto-batching VM.