Home
last modified time | relevance | path

Searched refs:ManagedPyObjects (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpython_ref_manager.h45 class ManagedPyObjects {
47 ManagedPyObjects() = default;
48 ManagedPyObjects(PythonRefManager* manager,
51 ~ManagedPyObjects();
53 ManagedPyObjects(const ManagedPyObjects& other) = delete;
54 ManagedPyObjects(ManagedPyObjects&& other) = default;
55 ManagedPyObjects& operator=(const ManagedPyObjects& other) = delete;
56 ManagedPyObjects& operator=(ManagedPyObjects&& other) = default;
66 std::shared_ptr<ManagedPyObjects> ManageReference(pybind11::object object);
67 std::shared_ptr<ManagedPyObjects> ManageReferences(
H A Dpython_ref_manager.cc24 PythonRefManager::ManagedPyObjects::ManagedPyObjects( in ManagedPyObjects() function in xla::PythonRefManager::ManagedPyObjects
33 PythonRefManager::ManagedPyObjects::~ManagedPyObjects() { in ~ManagedPyObjects()
39 std::shared_ptr<PythonRefManager::ManagedPyObjects>
41 return std::make_shared<ManagedPyObjects>(this, in ManageReference()
45 std::shared_ptr<PythonRefManager::ManagedPyObjects>
47 return std::make_shared<ManagedPyObjects>(this, objects); in ManageReferences()
H A Dpy_values.cc188 std::shared_ptr<PythonRefManager::ManagedPyObjects> py_buffer_ref = 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()