Wrapper for `Graph.stream()` using the default graph. See `tf.Graph.stream` for more details. Args: stream_idx: The index of gpu stream to use in the context. Returns: A context manager that specifies the default gpu stream to use for newly created ops.
(stream_idx=0)
| 5241 | |
| 5242 | @tf_export(v1=["stream"]) |
| 5243 | def stream(stream_idx=0): |
| 5244 | """Wrapper for `Graph.stream()` using the default graph. |
| 5245 | |
| 5246 | See `tf.Graph.stream` for more details. |
| 5247 | |
| 5248 | Args: |
| 5249 | stream_idx: The index of gpu stream to use in the context. |
| 5250 | |
| 5251 | Returns: |
| 5252 | A context manager that specifies the default gpu stream to use for newly |
| 5253 | created ops. |
| 5254 | """ |
| 5255 | return get_default_graph().stream(stream_idx) |
| 5256 | |
| 5257 | @tf_export(v1=["container"]) |
| 5258 | def container(container_name): |
nothing calls this directly
no test coverage detected