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.