tf.contrib.graph_editor.filter_ts
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 . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/contrib/graph_editor/filter_ts