Wrapper for `Graph.container()` using the default graph. 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.
(container_name)
| 5256 | |
| 5257 | @tf_export(v1=["container"]) |
| 5258 | def container(container_name): |
| 5259 | """Wrapper for `Graph.container()` using the default graph. |
| 5260 | |
| 5261 | Args: |
| 5262 | container_name: The container string to use in the context. |
| 5263 | |
| 5264 | Returns: |
| 5265 | A context manager that specifies the default container to use for newly |
| 5266 | created stateful ops. |
| 5267 | """ |
| 5268 | return get_default_graph().container(container_name) |
| 5269 | |
| 5270 | |
| 5271 | def _colocate_with_for_gradient(op, gradient_uid, ignore_existing=False): |