returnn.frontend.graph

Graph utils.

They might be just available for graph-based frameworks, but not necessarily.

returnn.frontend.graph.get_tensor_dependencies(x: Tensor) → Sequence[Tensor]

Parameters:

x – tensor

Returns:

list of tensors which are inputs to x

returnn.frontend.graph.get_tensor_consumers(x: Tensor) → Sequence[Tensor]

Parameters:

x – tensor

Returns:

list of tensors which consume x

returnn.frontend.graph.walk_tensor_consumers(seed: Tensor | Sequence[Tensor], *, filter_outputs: Callable[[Tensor], bool] | None = None, ending_condition: Callable[[Tensor], bool] | None = None) → List[Tensor]

Parameters:

• seed – tensor

• filter_outputs – if given, this function will be called with each tensor, and if it returns False, the tensor will be skipped.

• ending_condition – if given, this function will be called with each tensor, and if it returns True, we return.

Returns:

yields all tensors which consume seed, and their consumers, recursively