tf.contrib.graph_editor.filter_ts

View source on GitHub

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

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.

A list of tf.Tensor.

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