tf.container

View source on GitHub

Wrapper for Graph.container() using the default graph.

Aliases:

• tf.compat.v1.container

tf.container(container_name)

Args:

• container_name: The container string to use in the context.

Returns:

A context manager that specifies the default container to use for newly created stateful ops.