Get the layers those consumes this tensor.
(self, tensor)
| 405 | return state.name_to_layer.get(name, None) |
| 406 | |
| 407 | def get_tensor_users(self, tensor) -> Iterable["Layer"]: |
| 408 | ''' |
| 409 | Get the layers those consumes this tensor. |
| 410 | ''' |
| 411 | state = self._get_graph() |
| 412 | for layer in state.tensor_to_consumers[tensor]: |
| 413 | yield layer |
| 414 | |
| 415 | def get_tensor_parent(self, tensor) -> Optional["Layer"]: |
| 416 | ''' |