Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

Module: tfp.debugging.benchmarking

TensorFlow Probability benchmarking library.

Defined in python/debugging/benchmarking/__init__.py.

Classes

class BenchmarkTfFunctionConfig: BenchmarkTfFunctionConfig(strategies, hardware)

Functions

benchmark_tf_function(...): Time a TensorFlow function under a variety of strategies and hardware.

default_benchmark_config(...)

Other Members

RUNTIME_EAGER

RUNTIME_FUNCTION

RUNTIME_XLA

HARDWARE_CPU

HARDWARE_GPU