Missed TensorFlow World? Check out the recap. Learn more

tf.contrib.graph_editor.filter_ts

View source on GitHub

Get all the tensors which are input or output of an op in ops.

tf.contrib.graph_editor.filter_ts(
    ops,
    positive_filter
)

Args:

  • ops: an object convertible to a list of tf.Operation.
  • positive_filter: a function deciding whether to keep a tensor or not. If True, all the tensors are returned.

Returns:

A list of tf.Tensor.

Raises:

  • TypeError: if ops cannot be converted to a list of tf.Operation.