Return a read subgraph for this handle.
(graph, handle, dtype)
| 253 | |
| 254 | |
| 255 | def _get_handle_reader(graph, handle, dtype): |
| 256 | """Return a read subgraph for this handle.""" |
| 257 | graph_key = TensorHandle._get_reader_key(handle) |
| 258 | result = graph._handle_readers.get(graph_key) |
| 259 | if result is None: |
| 260 | # Create reader if we haven't done it. |
| 261 | handle_device = TensorHandle._get_device_name(handle) |
| 262 | with graph.as_default(), graph.device(handle_device): |
| 263 | holder = array_ops.placeholder(dtypes.string) |
| 264 | _register_handle_feeder(holder.graph, holder, dtype) |
| 265 | reader = gen_data_flow_ops.get_session_tensor(holder, dtype) |
| 266 | result = (holder, reader) |
| 267 | graph._handle_readers[graph_key] = result |
| 268 | return result |
| 269 | |
| 270 | |
| 271 | def _get_handle_mover(graph, feeder, handle): |
no test coverage detected