Home
last modified time | relevance | path

Searched refs:owning_pybuffer (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_values.h32 pybind11::object owning_pybuffer) in DevicePutResult()
33 : buffer(b), weak_type(weak_type), owning_pybuffer(owning_pybuffer) {} in DevicePutResult()
48 pybind11::object owning_pybuffer; member
H A Djax_jit.cc762 } else if (on_device.owning_pybuffer) { in CopyBuffersToDevice()
764 std::move(on_device.owning_pybuffer)); in CopyBuffersToDevice()
H A Dpy_client.cc267 return py::reinterpret_borrow<py::object>(put.owning_pybuffer); in BufferFromPyval()