Function _is_loop_edge

tensorflow/python/tpu/tensor_tracer_report.py:68–70

Returns true if the op is the end of a while-loop creating a cycle.

(op)

     list of nodes or the cycle of nodes found.
  """
  def _is_loop_edge(op):
    """Returns true if the op is the end of a while-loop creating a cycle."""
    return op.type in ['NextIteration']

  def _in_op_degree(op):
    """Returns the number of incoming edges to the given op.