Searched defs:py_buffer (Results 1 – 3 of 3) sorted by relevance
254 StatusOr<py::capsule> BufferToDLPackManagedTensor(py::handle py_buffer, in BufferToDLPackManagedTensor()
193 StatusOr<DevicePutResult> PyBufferHelper(py::handle obj, py::handle py_buffer, in PyBufferHelper()
456 auto py_buffer = absl::make_unique<PjRtStreamExecutorBuffer>( in AllocateDestinationBuffer() local