Return a deletion subgraph for this handle.
(graph, deleter_key, handle)
| 290 | |
| 291 | |
| 292 | def _get_handle_deleter(graph, deleter_key, handle): |
| 293 | """Return a deletion subgraph for this handle.""" |
| 294 | result = graph._handle_deleters.get(deleter_key) |
| 295 | if result is None: |
| 296 | # Create deleter if we haven't done it. |
| 297 | handle_device = TensorHandle._get_device_name(handle) |
| 298 | with graph.as_default(), graph.device(handle_device): |
| 299 | holder = array_ops.placeholder(dtypes.string) |
| 300 | deleter = gen_data_flow_ops.delete_session_tensor(holder) |
| 301 | result = (holder, deleter) |
| 302 | graph._handle_deleters[deleter_key] = result |
| 303 | return result |
no test coverage detected