tf.profiler.experimental.Profile

View source on GitHub

Context-manager profile API.

tf.profiler.experimental.Profile(
    logdir
)

Profiling will start when entering the scope, and stop and save the results to the logdir when exits the scope. Open TensorBoard profile tab to view results.

Example usage:

with tf.profiler.experimental.Profile("/path/to/logdir"):
  # do some work

Methods

__enter__

View source

__enter__()

__exit__

View source

__exit__(
    typ, value, tb
)