Stay organized with collections Save and categorize content based on your preferences.

Implements a two-program approach for optimizing summaries on TPU.

Inherits From: OptionalSummariesFunction

This version works with the result of create_tf_while_loop_fn.

function The underlying function to wrap.
**tf_function_kwargs Additional arguments to pass to tf.function.



View source

Call self as a function.