Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/python/
Dpython_ref_manager.h44 class ManagedPyObjects {
46 ManagedPyObjects() = default;
47 ManagedPyObjects(PythonRefManager* manager,
50 ~ManagedPyObjects();
52 ManagedPyObjects(const ManagedPyObjects& other) = delete;
53 ManagedPyObjects(ManagedPyObjects&& other) = default;
54 ManagedPyObjects& operator=(const ManagedPyObjects& other) = delete;
55 ManagedPyObjects& operator=(ManagedPyObjects&& other) = default;
65 std::shared_ptr<ManagedPyObjects> ManageReference(pybind11::object object);
66 std::shared_ptr<ManagedPyObjects> ManageReferences(
Dpython_ref_manager.cc24 PythonRefManager::ManagedPyObjects::ManagedPyObjects( in ManagedPyObjects() function in xla::PythonRefManager::ManagedPyObjects
33 PythonRefManager::ManagedPyObjects::~ManagedPyObjects() { in ~ManagedPyObjects()
42 std::shared_ptr<PythonRefManager::ManagedPyObjects>
44 return std::make_shared<ManagedPyObjects>(this, in ManageReference()
48 std::shared_ptr<PythonRefManager::ManagedPyObjects>
50 return std::make_shared<ManagedPyObjects>(this, objects); in ManageReferences()
Dpy_client.cc129 std::shared_ptr<PythonRefManager::ManagedPyObjects> py_buffer_ref = in PjRtBufferFromPyval()
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/
Dtpu_client_extension.cc112 std::shared_ptr<PythonRefManager::ManagedPyObjects> py_buffer_ref = in PYBIND11_MODULE()