Home
last modified time | relevance | path

Searched refs:py_buffer_ref (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.cc188 std::shared_ptr<PythonRefManager::ManagedPyObjects> py_buffer_ref = in HandleNumpyArray() local
191 [py_buffer_ref{ in HandleNumpyArray()
192 std::move(py_buffer_ref)}]() { /* keeps py_buffer_ref alive */ }; in HandleNumpyArray()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
H A Dtpu_client_extension.cc122 std::shared_ptr<PythonRefManager::ManagedPyObjects> py_buffer_ref = in PYBIND11_MODULE() local
134 std::move(leaves), tree.shape, std::move(py_buffer_ref), in PYBIND11_MODULE()