Save the date! Google I/O returns May 18-20 Register now

Module: tfp.debugging.benchmarking

TensorFlow Probability benchmarking library.

Classes

class BenchmarkTfFunctionConfig: BenchmarkTfFunctionConfig(strategies, hardware)

Functions

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

default_benchmark_config(...)

HARDWARE_CPU 'cpu'
HARDWARE_GPU 'gpu'
RUNTIME_EAGER 'eager'
RUNTIME_FUNCTION 'function/graph'
RUNTIME_XLA 'function/xla'
RUNTIME_XLA_AUTOCLUSTER 'xla-autoclustering'