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

from tensorflow.python import pywrap_tensorflow
from tensorflow.python.util.tf_export import tf_export


@tf_export('profiler.TraceMe')
class TraceMe(object):
  """Context manager that generates a trace event in the profiler."""

  def __init__(self, name):
    self._traceme = pywrap_tensorflow.PythonTraceMe(name)

  def __enter__(self):
    self._traceme.Enter()

  def __exit__(self, exc_type, exc_val, exc_tb):
    self._traceme.Exit()


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

from tensorflow.python import pywrap_tensorflow


class TraceMe(object):
  """Context manager that generates a trace event in the profiler."""

  def __init__(self, name):
    self._traceme = pywrap_tensorflow.PythonTraceMe(name)

  def __enter__(self):
    self._traceme.Enter()

  def __exit__(self, exc_type, exc_val, exc_tb):
    self._traceme.Exit()

Remove tf_export from TraceMe Python API.
Remove tf_export from TraceMe Python API.

PiperOrigin-RevId: 262247599

  
  from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  
  from tensorflow.python import pywrap_tensorflow
- from tensorflow.python.util.tf_export import tf_export
  
  
- @tf_export('profiler.TraceMe')
  class TraceMe(object):
    """Context manager that generates a trace event in the profiler."""
  
    def __init__(self, name):
      self._traceme = pywrap_tensorflow.PythonTraceMe(name)
  
    def __enter__(self):
      self._traceme.Enter()
  
    def __exit__(self, exc_type, exc_val, exc_tb):
      self._traceme.Exit()
  
Remove tf_export from TraceMe Python API.
## Code Before:

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

from tensorflow.python import pywrap_tensorflow
from tensorflow.python.util.tf_export import tf_export


@tf_export('profiler.TraceMe')
class TraceMe(object):
  """Context manager that generates a trace event in the profiler."""

  def __init__(self, name):
    self._traceme = pywrap_tensorflow.PythonTraceMe(name)

  def __enter__(self):
    self._traceme.Enter()

  def __exit__(self, exc_type, exc_val, exc_tb):
    self._traceme.Exit()

## Instruction:
Remove tf_export from TraceMe Python API.
## Code After:

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

from tensorflow.python import pywrap_tensorflow


class TraceMe(object):
  """Context manager that generates a trace event in the profiler."""

  def __init__(self, name):
    self._traceme = pywrap_tensorflow.PythonTraceMe(name)

  def __enter__(self):
    self._traceme.Enter()

  def __exit__(self, exc_type, exc_val, exc_tb):
    self._traceme.Exit()
