TensorFlow 2.0 Beta is available Learn more

tf.contrib.graph_editor.compute_boundary_ts

View source on GitHub

Compute the tensors at the boundary of a set of ops.

tf.contrib.graph_editor.compute_boundary_ts(ops)

This function looks at all the tensors connected to the given ops (in/out) and classify them into three categories: 1) input tensors: tensors whose generating operation is not in ops. 2) output tensors: tensors whose consumer operations are not in ops 3) inside tensors: tensors which are neither input nor output tensors.

Note that a tensor can be both an inside tensor and an output tensor if it is consumed by operations both outside and inside of ops.

Args:

  • ops: an object convertible to a list of tf.Operation.

Returns:

A tuple (outside_input_ts, outside_output_ts, inside_ts) where: outside_input_ts is a Python list of input tensors; outside_output_ts is a python list of output tensors; inside_ts is a python list of inside tensors. Since a tensor can be both an inside tensor and an output tensor, outside_output_ts and inside_ts might intersect.

Raises:

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