Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc86 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
93 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
119 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
123 GlobalPyRefManager()->ManageReferences( in PYBIND11_MODULE()
173 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
185 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
191 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
199 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/xla/python/
Dpython_ref_manager.h99 PythonRefManager* GlobalPyRefManager();
Dpython_ref_manager.cc85 PythonRefManager* GlobalPyRefManager() { in GlobalPyRefManager() function
Dcallback.h64 GlobalPyRefManager()->AddGarbage(absl::MakeSpan(&object, 1)); in ~CpuCallback()
Djax_jit.h46 xla::GlobalPyRefManager()->AddGarbage(absl::MakeSpan(&o, 1)); in ~JitState()
Dxla.cc151 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
157 GlobalPyRefManager()->CollectGarbage(); in PYBIND11_MODULE()
563 m.def("collect_garbage", []() { GlobalPyRefManager()->CollectGarbage(); }); in PYBIND11_MODULE()
Dpy_buffer.cc201 GlobalPyRefManager()->CollectGarbage(); in CopyToDevice()
235 GlobalPyRefManager()->CollectGarbage(); in BlockHostUntilReady()
601 GlobalPyRefManager()->CollectGarbage(); in RegisterTypes()
Dtraceback.cc108 GlobalPyRefManager()->AddGarbage(traceback.frames_); in SafeDestroy()
Ddlpack.cc58 GlobalPyRefManager()->AddGarbage( in ~DLPackTensor()
Dpy_values.cc189 GlobalPyRefManager()->ManageReference(std::move(array)); in HandleNumpyArray()
Dpy_client.cc252 GlobalPyRefManager()->CollectGarbage(); in BufferFromPyval()
Djax_jit.cc856 xla::GlobalPyRefManager()->MaybeCollectGarbage(); in Call()