Exemplos de TFE_Py_TapeSetStopOnThread em Python

Exemplo n.º 1

def stop_recording():
  """Stop all gradient recording (backprop and forwardprop)."""
  is_stopped = pywrap_tensorflow.TFE_Py_TapeSetIsStopped()
  try:
    if not is_stopped:
      pywrap_tensorflow.TFE_Py_TapeSetStopOnThread()
    yield
  finally:
    if not is_stopped:
      pywrap_tensorflow.TFE_Py_TapeSetRestartOnThread()

Exemplo n.º 2

def stop_recording():
    try:
        pywrap_tensorflow.TFE_Py_TapeSetStopOnThread()
        yield
    finally:
        pywrap_tensorflow.TFE_Py_TapeSetRestartOnThread()