Home
last modified time | relevance | path

Searched refs:GlobalPyRefManager (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc76 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
83 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
109 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
113 GlobalPyRefManager()->ManageReferences( in PYBIND11_MODULE()
148 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
155 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
163 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc131 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
137 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
302 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
388 m.def("collect_garbage", []() { GlobalPyRefManager()->CollectGarbage(); }); in PYBIND11_MODULE()
Dpython_ref_manager.h86 PythonRefManager* GlobalPyRefManager();
Dtraceback.cc33 GlobalPyRefManager()->AddGarbage(frames_); in ~Traceback()
Dpython_ref_manager.cc81 PythonRefManager* GlobalPyRefManager() { in GlobalPyRefManager() function
Dpy_buffer.cc74 GlobalPyRefManager()->CollectGarbage(); in CopyToDevice()
86 GlobalPyRefManager()->CollectGarbage(); in BlockHostUntilReady()
Dpy_client.cc117 GlobalPyRefManager()->CollectGarbage(); in PjRtBufferFromPyval()
130 GlobalPyRefManager()->ManageReference(std::move(c->array)); in PjRtBufferFromPyval()
Ddlpack.cc56 GlobalPyRefManager()->AddGarbage( in ~DLPackTensor()