Home
last modified time | relevance | path

Searched refs:TFE_Py_TapeSetStopOnThread (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dtape.py156 pywrap_tfe.TFE_Py_TapeSetStopOnThread()
H A Dpywrap_tfe.h189 void TFE_Py_TapeSetStopOnThread();
H A Dpywrap_tfe_src.cc1915 void TFE_Py_TapeSetStopOnThread() { *ThreadTapeIsStopped() = true; } in TFE_Py_TapeSetStopOnThread() function
/aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/
H A Dsymbols_pybind.txt176 TFE_Py_TapeSetStopOnThread
/aosp_15_r20/external/tensorflow/tensorflow/python/
H A Dtfe_wrapper.cc1048 m.def("TFE_Py_TapeSetStopOnThread", &TFE_Py_TapeSetStopOnThread); in PYBIND11_MODULE()