Module: tfp.experimental.auto_batching.numpy_backend

View source on GitHub

Numpy backend for auto-batching VM.

It can be faster than TF for tiny examples and prototyping, and moderately simpler due to immediate as opposed to deferred result computation.

All operations take and ignore name= arguments to allow for useful op names in the TensorFlow backend.

Classes

class NumpyBackend: Implements the Numpy backend ops for a PC auto-batching VM.