Home
last modified time | relevance | path

Searched refs:PythonHooks (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/profiler/internal/
Dpython_hooks.cc115 /*static*/ PythonHookContext* PythonHooks::e2e_context_ = nullptr;
127 PythonHooks* PythonHooks::GetSingleton() { in GetSingleton()
128 static PythonHooks* singleton = new PythonHooks; in GetSingleton()
163 PythonHooks* singleton = PythonHooks::GetSingleton(); in Start()
169 PythonHooks::set_e2e_context(e2e_context.release()); in Start()
243 /*static*/ int PythonHooks::ProfileFunction(PyObject* obj, PyFrameObject* frame, in ProfileFunction()
249 void PythonHooks::ProfileSlow(const py::object& frame, const string& event, in ProfileSlow()
355 PythonHooks* singleton = PythonHooks::GetSingleton(); in SetProfilerInAllThreads()
361 PyEval_SetProfile(&PythonHooks::ProfileFunction, nullptr); in SetProfilerInAllThreads()
371 PyEval_SetProfile(&PythonHooks::ProfileFunction, nullptr); in SetProfilerInAllThreads()
Dpython_hooks.h110 class PythonHooks; variable
116 friend class ::tensorflow::profiler::PythonHooks;
140 class PythonHooks {
142 static PythonHooks* GetSingleton();
/external/tensorflow/tensorflow/core/profiler/backends/cpu/
Dpython_tracer.cc62 PythonHooks::GetSingleton()->Start(options_); in Start()
71 context_ = PythonHooks::GetSingleton()->Stop(); in Stop()