Home
last modified time | relevance | path

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

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