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 .
|