mlx.core.depends

Contents

mlx.core.depends#

depends(inputs: array | Sequence[array], dependencies: array | Sequence[array])#

Insert dependencies between arrays in the graph. The outputs are identical to inputs but with dependencies on dependencies.

Parameters:
  • inputs (array or Sequence[array]) – The input array or arrays.

  • dependencies (array or Sequence[array]) – The array or arrays to insert dependencies on.

Returns:

The outputs which depend on dependencies.

Return type:

array or Sequence[array]