"""Ops for pfor, for_loop, jacobian."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python.ops.parallel_for import *  # pylint: disable=wildcard-import
from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
from tensorflow.python.ops.parallel_for.gradients import jacobian
from tensorflow.python.util.all_util import remove_undocumented

_allowed_symbols = [
    'pfor',
    'for_loop',
    'jacobian',
    'batch_jacobian',
]

remove_undocumented(__name__, _allowed_symbols)

"""Ops for pfor, for_loop, jacobian."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python.ops.parallel_for import *  # pylint: disable=wildcard-import
from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
from tensorflow.python.ops.parallel_for.gradients import jacobian

Remove usage of remove_undocumented from core parallel_for.
Remove usage of remove_undocumented from core parallel_for.

remove_undocumented is causing issues with our pip tests.
remove_undocumented is not used anywhere else in core TF code
and we have a new mechanism for annotating the public TF API.

  """Ops for pfor, for_loop, jacobian."""
  
  from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  
  from tensorflow.python.ops.parallel_for import *  # pylint: disable=wildcard-import
  from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
  from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
  from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
  from tensorflow.python.ops.parallel_for.gradients import jacobian
- from tensorflow.python.util.all_util import remove_undocumented
  
- _allowed_symbols = [
-     'pfor',
-     'for_loop',
-     'jacobian',
-     'batch_jacobian',
- ]
- 
- remove_undocumented(__name__, _allowed_symbols)
- 
Remove usage of remove_undocumented from core parallel_for.
## Code Before:
"""Ops for pfor, for_loop, jacobian."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python.ops.parallel_for import *  # pylint: disable=wildcard-import
from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
from tensorflow.python.ops.parallel_for.gradients import jacobian
from tensorflow.python.util.all_util import remove_undocumented

_allowed_symbols = [
    'pfor',
    'for_loop',
    'jacobian',
    'batch_jacobian',
]

remove_undocumented(__name__, _allowed_symbols)

## Instruction:
Remove usage of remove_undocumented from core parallel_for.
## Code After:
"""Ops for pfor, for_loop, jacobian."""

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from tensorflow.python.ops.parallel_for import *  # pylint: disable=wildcard-import
from tensorflow.python.ops.parallel_for.control_flow_ops import for_loop
from tensorflow.python.ops.parallel_for.control_flow_ops import pfor
from tensorflow.python.ops.parallel_for.gradients import batch_jacobian
from tensorflow.python.ops.parallel_for.gradients import jacobian
