Missed TensorFlow World? Check out the recap. Learn more

Module: tfp.debugging.benchmarking

View source on GitHub

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(...)

Other Members

  • HARDWARE_CPU = 'cpu'
  • HARDWARE_GPU = 'gpu'
  • RUNTIME_EAGER = 'eager'
  • RUNTIME_FUNCTION = 'function/graph'
  • RUNTIME_XLA = 'function/xla'