def _tfr_control_flow_nested_if(x, y, z, select):
  if select == 'x':
    return x
  elif select == 'y':
    return y
  else:
    return z


@composite.Composite('TestInputNOp')
def _tfr_control_flow_range_for(x):
  # TODO(fengliuai): use len(x) instead
  n = 10
  x_sum = x[0]
  for i in range(1, n):
    x_sum = math_ops.Add(x_sum, x[i])
  return x_sum


@composite.Composite('TestInputNOp')