TensorFlow 2.0 Beta is available Learn more

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