Home
last modified time | relevance | path

Searched defs:py_buffer (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Ddlpack.cc266 StatusOr<py::capsule> BufferToDLPackManagedTensor(py::handle py_buffer, in BufferToDLPackManagedTensor()
H A Dpy_values.cc206 StatusOr<DevicePutResult> PyBufferHelper(py::handle obj, py::handle py_buffer, in PyBufferHelper()
H A Dpy_client.cc132 PyBuffer* py_buffer; in Defragment() member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dpjrt_stream_executor_client.cc466 auto py_buffer = std::make_unique<PjRtStreamExecutorBuffer>( in AllocateDestinationBuffer() local